用yolo5官方pt模型进行转换,yolov5示例检测结果出错
收藏回复举报
用yolo5官方pt模型进行转换,yolov5示例检测结果出错
t('forum.solved') 已解决
新人帖
发表于2023-12-01 19:14:12
0 查看

这是设备安装的镜像版本

cke_3921.png

安装yolov5

git clone https://github.com/ultralytics/yolov5
cd yolov5
git checkout v6.0
pip install -r requirements.txt
pip install -U coremltools onnx scikit-learn==0.19.2

我从官方下载了pt模型

https://github.com/ultralytics/yolov5/releases/download/v6.0/yolov5s.pt

pt转onnx

python export.py --include onnx --weights yolov5s.pt --img 640 --batch 1 --opset=11

onnx 转om

atc --model=yolo5s.onnx --framework=5 --output=model --soc_version=Ascend310B4

运行notebooks/01-yolov5/ 示例结果是这样的 :

cke_24909.png

识别已经混乱了,是什么原因   

我要发帖子