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

时间:2024-05-28 16:22:36

Step5 推理

  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. 运行完毕后,在当前目录的export/exp_tmp下可以看到推理后的文件,打开后显示如下。
    图2 推理后的文件
support.huaweicloud.com/devtool-modelarts/devtool-modelarts_0178.html