铜陵商城网站建设,广州门户网站建设方案,wordpress 附件清理,数字展厅网站建设深度学习目标检测ui界面-交通标志检测识别
为了将算法封装起来#xff0c;博主尝试了实验pyqt5的上位机界面进行封装#xff0c;其中遇到了一些坑举给大家避开。这里加载的训练模型参考之前写的博客#xff1a; 自动驾驶目标检测项目实战(一)—基于深度学习框架yolov的交通…深度学习目标检测ui界面-交通标志检测识别
为了将算法封装起来博主尝试了实验pyqt5的上位机界面进行封装其中遇到了一些坑举给大家避开。这里加载的训练模型参考之前写的博客 自动驾驶目标检测项目实战(一)—基于深度学习框架yolov的交通标志检测
效果 输入设置好账号密码 加载一张交通标志图片 点击开始 测试使用cpu跑的使用gpu的话检测速度会更快。
过程
主要包括检测代码和界面代码 我们只需要将检测完的图片在界面显示即可但是这样遇到一些问题
1QtGui.QImage加载图片时图片的红色变成蓝色解决方法一开始猜想是图像通道问题于是把 _image QtGui.QImage(self.img_defect[:], self.img_defect.shape[1], self.img_defect.shape[0],self.img_defect.shape[1] * 3,QtGui.QImage.Format_RGB888) 中的Format_RGB888改为Format_BGR888即 _image QtGui.QImage(self.img_defect[:], self.img_defect.shape[1], self.img_defect.shape[0],self.img_defect.shape[1] * 3,QtGui.QImage.Format_BGR888) 就可以正常显示图片的颜色。
2另外如何使用以下代码直接读取截图的png图片时图片会乱码 img_orignal cv_imread(self.fname) # opencv读取图片res cv2.resize(img_orignal, (200, 200), interpolationcv2.INTER_CUBIC) # 用cv2.resize设置图片大小self.img_o res_image QtGui.QImage(self.img_o[:], self.img_o.shape[1], self.img_o.shape[0], self.img_o.shape[1] * 3,QtGui.QImage.Format_BGR888) # pyqt5转换成自己能放的图片格式jpg_out QtGui.QPixmap(_image) # 转换成QPixmapself.label_4.setPixmap(jpg_out) # 设置图片显示解决方法 先用opencv保存图片如何再加载 # 显示图片img_orignal cv_imread(self.fname) # opencv读取图片cv2.imwrite(./images/res/origin.jpg, img_orignal)img cv_imread(./images/res/origin.jpg)res cv2.resize(img, (200, 200), interpolationcv2.INTER_CUBIC) # 用cv2.resize设置图片大小self.img_o res_image QtGui.QImage(self.img_o[:], self.img_o.shape[1], self.img_o.shape[0], self.img_o.shape[1] * 3,QtGui.QImage.Format_BGR888) # pyqt5转换成自己能放的图片格式jpg_out QtGui.QPixmap(_image) # 转换成QPixmapself.label_4.setPixmap(jpg_out) # 设置图片显示就可以解决 当然这只是其中一种解决方法其他我还没发现。
记录到这里需要源码的私信
