您现在的位置是:首页 >技术交流 >yolov5+pyqt5(GUI界面,可更换自己的模型)网站首页技术交流

yolov5+pyqt5(GUI界面,可更换自己的模型)

宇宙无敌霹雳暴暴龙战士 2023-05-29 08:00:02
简介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就可以出现自己的界面了。程序里还有保存图片,更换背景啥的可以自己尝试换一下。

示例图:

 

风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。