Javacr / PyQt5-YOLOv5

PyQt5 implementation of YOLOv5 GUI
GNU General Public License v3.0
1.3k stars 220 forks source link

怎么运行自己的模型呢?只需要更改pt文件吗? #31

Open dragondcx opened 1 year ago

WesleyYeh commented 1 year ago

同求,我添加了自己的pt文件,但是运行界面左下角会报错 Can't get attribute 'DetectionModel' on <module 'models.yolo' from 'F:\Pqt.._v6.1\models\yolo.py'>

faithzxr commented 1 year ago

@dragondcx 你解决了嘛?怎么换成自己的模型?

WesleyYeh commented 1 year ago

@dragondcx 你解决了嘛?怎么换成自己的模型?

需要用yolov5_6.1的版本训练出来的模型,别的yolov5版本的会报错

faithzxr commented 1 year ago

@WinsYe 就是只用他的UI界面,模型换成自己的,怎么改呀?

WesleyYeh commented 1 year ago

直接把自己的pt文件 放到他的路径里就好了啊

------------------ 原始邮件 ------------------ 发件人: @.>; 发送时间: 2023年4月24日(星期一) 下午5:37 收件人: @.>; 抄送: @.>; @.>; 主题: Re: [Javacr/PyQt5-YOLOv5] 怎么运行自己的模型呢?只需要更改pt文件吗? (Issue #31)

@WinsYe 就是只用他的UI界面,模型换成自己的,怎么改呀?

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>

faithzxr commented 1 year ago

@WinsYe 现在有模型了,但是和它训练的yolov5的版本不太一样,应该怎么处理呢?

WesleyYeh commented 1 year ago

@dragondcx 你解决了嘛?怎么换成自己的模型?

需要用yolov5_6.1的版本训练出来的模型,别的yolov5版本的会报错

需要用yolov5_6.1的版本训练

faithzxr commented 1 year ago

好的,谢谢你啦 @WinsYe

Fortunate-ziye commented 6 months ago

可以看一下这个,还有一些问题,但至少不会报这个错了 https://blog.csdn.net/weixin_47037450/article/details/131912962