search

airockchip / rknn_model_zoo

Apache License 2.0
1.05k stars 192 forks source link

关于YOLOV5模型训练的问题 #67

Open pan1024 opened 10 months ago

pan1024 commented 10 months ago

是否使用任意版本YOLOV5源码训练的得到的pt文件，再使用该版本携带的export.py转为ONNX即可，还是有特定版本需求以及export.py修改需求？

lucameng commented 8 months ago

我用过最新版的yolov5模型,训练并转化得到的模型有问题.换回旧版的模型,无问题发生.

Starkguava commented 1 month ago

旧模型的输出是12552020 12554040 12558080三个 新模型是125200*（类别数量+5）有没有相关更改和demo？