运行Open Model Zoo多相机多目标跟踪Demo遇模型格式不兼容问题,求ONNX格式模型重新下载方法
运行Open Model Zoo多相机多目标跟踪Demo遇模型格式不兼容问题,求ONNX格式模型重新下载方法
我来帮你解决这个问题~从报错信息来看,你当前使用的是Open Model Zoo(OMZ)默认下载的IR格式模型(.xml/.bin),但这个Demo的推理后端只支持ONNX、PyTorch等格式,所以才会出现不兼容的错误。下面是具体的解决步骤:
1. 下载指定模型的ONNX格式版本
你之前用omz_downloader --all下载的是所有模型的IR格式版本,我们可以单独指定格式来下载目标模型的ONNX版本:
- 打开终端,运行以下命令下载person-detection-retail-0013的ONNX格式模型:
omz_downloader --name person-detection-retail-0013 --output_dir ./models --format onnx - 同样的,下载person-reidentification-retail-0277的ONNX格式模型:
这里的omz_downloader --name person-reidentification-retail-0277 --output_dir ./models --format onnx--output_dir可以指定你想要存放模型的目录,方便后续查找。
2. 调整Demo运行命令
下载完成后,需要修改Demo的启动命令,把模型路径换成ONNX文件的实际路径。比如如果模型下载到了./models/intel/目录下,命令应该改成:
python3.9 multi_camera_multi_target_tracking_demo.py -i ./test_video/test1.mp4 ./test_video/test1.mp4 --m_detector ./models/intel/person-detection-retail-0013/onnx/person-detection-retail-0013.onnx --m_reid ./models/intel/person-reidentification-retail-0277/onnx/person-reidentification-retail-0277.onnx
3. 特殊情况处理(直接下载ONNX失败时)
如果遇到omz_downloader提示没有ONNX格式的模型,你可以先下载IR格式,再用omz_converter转换为ONNX:
- 先下载IR格式模型:
omz_downloader --name person-detection-retail-0013 --output_dir ./models - 转换为ONNX格式:
重复这个步骤处理重识别模型即可。omz_converter --name person-detection-retail-0013 --input_dir ./models/intel/person-detection-retail-0013/FP32 --output_dir ./models/intel/person-detection-retail-0013/onnx --format onnx
这样应该就能解决模型格式不兼容的问题啦~
备注:内容来源于stack exchange,提问作者bbartling
相关产品推荐
相关产品推荐

