AI开发平台MODELARTS-使用自定义数据集进行训练:Step5 推理

时间：2024-05-17 17:20:04

AI开发平台MODELARTS 使用样例

Step5 推理

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

在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