前提 YOLO模型已经训练好,若还没训练,可参考我的第一篇博客博客 一、生成onnx文件 yolov7目前可以直接通过其自身带的export.py导出onnx模型,过程中出现的问题大多都是包的问题,如果没有哪个包,就在相应的环境中下载即可。其中需要注意的是coremltools这个包的下载版本。 我首先是将coremltools这个包按要求下载为4.1然后会报错,又将包升级到5.2后还是会报错,最后几经挫折在GitHub上找到原因,解决方法是下载5.0b版本就好了。 二、vs2019调用ONNX文件 参考博客博客 需要注意的是:在yolo.h的文件下需要改变类别的值