search

qqsuhao / YOLOv3-YOLOv3-tiny-yolo-fastest-xl--pytorch

YOLOv3/YOLOv3-tiny/yolo-fasetest-xl从训练到部署
MIT License
22 stars 9 forks source link

使用modelv5，将配置文件改为yolo'-fastest-xl ，训练报错，在网络permute时报形状不匹配 #1

Open omtrix opened 3 years ago

qqsuhao commented 3 years ago

modelv5.py是可以运行并生成onnx文件的，你可以先尝试使用yolo-fastest-xl执行modelv5.py判断一下是不是这个py文件的错误。如果不是这个文件的错误，那很有可能是train.py的参数与yolo-fastest-xl.cfg中的参数不一致，比如输入图像的大小，数据集种类的个数等。或者你可以尝试调试一下。

发送自 Windows 10 版邮件https://go.microsoft.com/fwlink/?LinkId=550986应用

发件人: @.> 发送时间: 2021年7月6日 10:07 收件人: @.> 抄送: @.***> 主题: [qqsuhao/YOLOv3-YOLOv3-tiny-yolo-fastest-xl--pytorch] 使用modelv5，将配置文件改为yolo'-fastest-xl ，训练报错，在网络permute时报形状不匹配 (#1)

― You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHubhttps://github.com/qqsuhao/YOLOv3-YOLOv3-tiny-yolo-fastest-xl--pytorch/issues/1, or unsubscribehttps://github.com/notifications/unsubscribe-auth/AHSNB6QZFOSTXNGIDOOMBS3TWJQNFANCNFSM473TXLLA.

omtrix commented 3 years ago

我找到原因了，我cfg文件没有完全改完，还需要将全连接层的filters=255 改成自己训练类别所对应的数量，谢谢指教