在执行模型推理的目标跟踪案例时,到最后一步执行推理时报错!目标检测方面不太懂,请问这个bug如何解决?
收藏回复举报
在执行模型推理的目标跟踪案例时,到最后一步执行推理时报错!目标检测方面不太懂,请问这个bug如何解决?
t('forum.solved') 已解决
发表于2023-07-13 15:41:18
0 查看

python3 om_infer.py --model=output/yolov5s_bs1.om --eval --visible


(base) root@davinci-mini:/home/infer_project# python3 om_infer.py --model=output/yolov5s_bs1.om --eval --visible

/root/miniconda3/lib/python3.9/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension:

  warn(f"Failed to load image Python extension: {e}")

[INFO] acl init success

[INFO] open device 0 success

[INFO] load model output/yolov5s_bs1.om success

[INFO] create model description success

100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5/5 [00:09<00:00,  1.89s/it]

Average infer time: 27.684 ms / 1 image

saveing predictions.json to output/

Start evaluate *bbox* results...

loading annotations into memory...

Done (t=0.00s)

creating index...

index created!

Loading and preparing results...

Traceback (most recent call last):

  File "/home/infer_project/om_infer.py", line 249, in <module>

    main()

  File "/home/infer_project/om_infer.py", line 229, in main

    eval(args.ground_truth_json, pred_json_path)

  File "/home/infer_project/om_infer.py", line 115, in eval

    cocoDt = cocoGt.loadRes(cocoDt_file)

  File "/root/miniconda3/lib/python3.9/site-packages/pycocotools/coco.py", line 329, in loadRes

    if 'caption' in anns[0]:

IndexError: list index out of range

[INFO] unload model success, model Id is 1

[INFO] end to destroy context

[INFO] end to reset device is 0

[INFO] end to finalize acl

我要发帖子