使用现有模型进行推理

Step0 准备环境

使用前需要先准备开发环境。

1. 参考准备开发环境章节创建并打开Notebook实例。
2. 参考创建算法工程章节创建算法工程。

Step1 安装算法套件

安装ivgPose算法套件。

python manage.py install algorithm ivgPose

Step2 安装预训练模型

默认下载到当前工程的./model_zoo/{算法类型}/{算法版本}目录下。

python manage.py install model ivgPose:body/simplepose_resnet50_coco_256x192

Step3 推理

1. 准备好待推理的图片，本小节以算法套件里内置的推理图片为例，原图片如下。
   图1 待推理的图片
   

2. 在Terminal中执行下述推理命令，其中--img_file的值为待推理图片的路径。

   python manage.py run --cfg algorithms/ivgPose/config/sample/config.py --pipeline infer --demo_type image_demo --load_from model_zoo/ivgPose/body/simplepose_resnet50_coco_256x192.pth.tar --img_path algorithms/ivgPose/algorithm/examples/images/body/human36m_s_01_act_02_subact_01_ca_01_000001.jpg  --is_show
   

3. 运行完毕后，在当前目录的output/simplepose_resnet50_coco_256x192/下可以看到推理后的文件human36m_s_01_act_02_subact_01_ca_01_000001_vis.jpg，打开后显示如下。
   图2 推理后的文件