您现在的位置是:首页 >技术交流 >yolov5+pyqt5(GUI界面,可更换自己的模型)网站首页技术交流
yolov5+pyqt5(GUI界面,可更换自己的模型)
接上两篇,在改进完yolov5之后需要做个GUI界面进行屎上插画。
我用的是景唯up的代码,但是因为版本问题或者别的其他什么乱七八糟的问题,界面不能显示。没有label的图片可以显示,但是有label的图片会在左下角注脚出出现label名称,但是原图和标记图都不会显示。
景唯大大源码:yolov5+pyqt5
可以先尝试一下能不能用,就把pt文件夹里的pt文件换成自己训练好的模型就行,如果能用,就什么都不用改,如果不能用,并且和我一样问题的,可以看看我的过程。
我的程序对common.py和yolo.py做了修改,也就是更换了主体函数。
1.首先就是移植,我和原作者虽然都用的yolov5但是版本不一样,导致部分函数大同小异,可以对比一下自己的general.py和作者用的是不是同一版本。复制pyqt5文件夹中的所有文件,粘贴到自己yolov5文件夹里,遇到相同的就跳过该操作,不要漏了。(每一个文件夹都打开看看!!!)
2.把训练好的best.pt放到pt文件夹里,运行一下试试能不能出图。
3.如果不能出图,打开main.py文件,ctr+f搜索statistic,在这句话前后加上俩print,再次运行,如果是有a无b,那就是和我一样的问题。
4.再次ctr+f搜索names,第一个names,在这下面打印names
如果出现是0:label1,1:label2.....这种字典形式,就是label的转换问题
再次ctr+f定位statistic-dic这一行,在前面把自己的label重新写到一个空的列表里,我用的是names1,然后把后面那个框框也改成names1。(记得删除前面添加的print,print只是为了找出程序运行的断点)
然后再重新运行main.py就可以出现自己的界面了。程序里还有保存图片,更换背景啥的可以自己尝试换一下。
示例图: