示例#1
0
    def predict_frame(self, imgPath):
        Eng2Chi = {"green": "绿色", "blue": "蓝色", "yellow": "黄色"}
        q = Predict()
        afterprocess, old = q.preprocess(imgPath)
        colors, card_imgs = q.locate_carPlate(afterprocess, old)
        result, roi, color, divs = q.char_recogize(colors,
                                                   card_imgs)  # all list
        if len(result) == 0:
            print("未能识别到车牌")
            self.colorLabel.setText("抱歉未能识别到车牌")
            self.NumLabel.setText("抱歉,未能识别到车牌")
        else:
            for r in range(len(result)):
                print("#" * 10 + "识别结果是" + "#" * 10)
                print("车牌的颜色为:" + Eng2Chi[color[r]])
                self.colorLabel.setText(Eng2Chi[color[r]])
                final_color = Eng2Chi[color[r]]
                if final_color == "蓝色":
                    self.colorLabel.setStyleSheet(
                        "QLabel{background-color:blue;color:white;font: 20pt;alignment:center;}"
                    )
                if final_color == "黄色":
                    self.colorLabel.setStyleSheet(
                        "QLabel{background-color:yellow;color:black;font: 20pt;alignment:center;}"
                    )
                if final_color == "绿色":
                    self.colorLabel.setStyleSheet(
                        "QLabel{background-color:green;color:white;font: 20pt;alignment:center;}"
                    )
                self.colorLabel.setText(final_color)
                print(result[r])
                result[r].insert(2, "-")
                final_result = ''.join(result[r])
                self.NumLabel.setText(''.join(result[r]))
                print("#" * 25)
                # roi[r] = cv2.cvtColor(roi[r], cv2.COLOR_BGR2RGB)
                # QtImg = QtGui.QImage(roi[r].data, roi[r].shape[1], roi[r].shape[0], QtGui.QImage.Format_RGB888)
                # # 显示图片到label中
                # self.location.resize(QtCore.QSize(roi[r].shape[1], roi[r].shape[0]))
                # self.location.setPixmap(QtGui.QPixmap.fromImage(QtImg))

                if len(final_result
                       ) >= 8 and u'\u4e00' <= final_result[0] <= u'\u9fff':
                    #保存至MYSQL数据库
                    conn, cur = self.connetSQL()
                    currenttime = datetime.datetime.now()
                    print(currenttime.strftime('%Y-%m-%d %H:%M:%S'))
                    try:
                        cur.execute(
                            "insert into plate(plate_num,plate_color,time)values(%s,%s,%s)",
                            (final_result, final_color, currenttime))
                        conn.commit()
                    except Exception as e:
                        print("expect: ", e)
                    finally:
                        cur.close()
                        conn.close()
示例#2
0
    def openimage(self):
        Eng2Chi = {"green": "绿色", "blue": "蓝色", "yellow": "黄色"}
        #打开文件路径
        #设置文件扩展名过滤,注意用双分号间隔
        imgPath, imgType = QFileDialog.getOpenFileName(
            self, "打开图片", "", "*.jpg;;*.png;;*.jpeg;;*.bmp;;All Files (*)")
        print(imgPath)
        #利用qlabel显示图片
        png = QtGui.QPixmap(imgPath).scaled(self.imgLabel.width(),
                                            self.imgLabel.height())
        self.imgLabel.setPixmap(png)

        if imgPath:
            ###################识别#####################
            final_result = None
            color_index = None
            q = Predict()
            afterprocess, old = q.preprocess(imgPath)
            self.Img_preprocess = afterprocess
            colors, card_imgs = q.locate_carPlate(afterprocess, old)
            result, roi, color, divs = q.char_recogize(colors,
                                                       card_imgs)  # all list
            print("识别到了{0}个车牌".format(len(result)))
            if len(result) == 0:
                print("未能识别到车牌")
                self.colorLabel.setText("抱歉未能识别到车牌")
                self.NumLabel.setText("抱歉,未能识别到车牌")
                self.location.clear()
            else:
                for r in range(len(result)):
                    if len(result[r]) >= 7:
                        final_result = result[r]
                        color_index = r
                        findex = r

                if not final_result or len(final_result) < 7:
                    self.NumLabel.setText("抱歉,未能识别到车牌")
                    return

                print("#" * 10 + "识别结果是" + "#" * 10)
                ########车牌颜色########
                print("车牌的颜色为:" + Eng2Chi[color[color_index]])
                final_color = Eng2Chi[color[color_index]]
                if final_color == "蓝色":
                    self.colorLabel.setStyleSheet(
                        "QLabel{background-color:blue;color:white;font: 20pt;alignment:center;}"
                    )
                if final_color == "黄色":
                    self.colorLabel.setStyleSheet(
                        "QLabel{background-color:yellow;color:black;font: 20pt;alignment:center;}"
                    )
                if final_color == "绿色":
                    self.colorLabel.setStyleSheet(
                        "QLabel{background-color:green;color:white;font: 20pt;alignment:center;}"
                    )
                self.colorLabel.setText(final_color)
                ########车牌颜色########

                ########车牌号码########
                final_result.insert(2, "-")
                result = ''.join(final_result)
                print(result)
                self.NumLabel.setText(result)
                print("#" * 25)
                roi[color_index] = cv2.cvtColor(roi[color_index],
                                                cv2.COLOR_BGR2RGB)
                qimg = qimage2ndarray.array2qimage(roi[color_index])
                local_img = qimg.scaled(self.location.width(),
                                        self.location.height())
                self.location.setPixmap(QtGui.QPixmap(local_img))
                # QtImg = QtGui.QImage(roi[r].data,roi[r].shape[1],roi[r].shape[0],QtGui.QImage.Format_RGB888)
                # #显示图片到label中
                # self.location.resize(QtCore.QSize(roi[r].shape[1],roi[r].shape[0]))
                # self.location.setPixmap(QtGui.QPixmap.fromImage(QtImg))
                ########车牌号码########

                if len(final_result
                       ) >= 8 and u'\u4e00' <= final_result[0] <= u'\u9fff':
                    #保存至MYSQL数据库
                    conn, cur = self.connetSQL()
                    currenttime = datetime.datetime.now()
                    print(currenttime.strftime('%Y-%m-%d %H:%M:%S'))
                    try:
                        cur.execute(
                            "insert into plate(plate_num,plate_color,time)values(%s,%s,%s)",
                            (final_result, final_color, currenttime))
                        conn.commit()
                    except Exception as e:
                        print("expect: ", e)
                    finally:
                        cur.close()
                        conn.close()

            ###################识别#####################

            if len(divs) == 0:
                pass
            elif len(divs[0]) < 7:
                pass
            else:
                # 与上同理
                Gray1 = self.QImage2Pixmap(0, divs)
                #self.div1.resize(QtCore.QSize(divs[0][0].shape[1], divs[0][0].shape[0]))
                Gray1 = Gray1.scaled(self.div1.width(), self.div1.height())
                self.div1.setPixmap(QtGui.QPixmap.fromImage(Gray1))

                Gray2 = self.QImage2Pixmap(1, divs)
                # self.div1.resize(QtCore.QSize(divs[0][0].shape[1], divs[0][0].shape[0]))
                Gray2 = Gray2.scaled(self.div2.width(), self.div2.height())
                self.div2.setPixmap(QtGui.QPixmap.fromImage(Gray2))

                Gray3 = self.QImage2Pixmap(2, divs)
                # self.div1.resize(QtCore.QSize(divs[0][0].shape[1], divs[0][0].shape[0]))
                Gray3 = Gray3.scaled(self.div3.width(), self.div3.height())
                self.div3.setPixmap(QtGui.QPixmap.fromImage(Gray3))

                Gray4 = self.QImage2Pixmap(3, divs)
                # self.div1.resize(QtCore.QSize(divs[0][0].shape[1], divs[0][0].shape[0]))
                Gray4 = Gray4.scaled(self.div4.width(), self.div4.height())
                self.div4.setPixmap(QtGui.QPixmap.fromImage(Gray4))

                Gray5 = self.QImage2Pixmap(4, divs)
                # self.div1.resize(QtCore.QSize(divs[0][0].shape[1], divs[0][0].shape[0]))
                Gray5 = Gray5.scaled(self.div5.width(), self.div5.height())
                self.div5.setPixmap(QtGui.QPixmap.fromImage(Gray5))

                Gray6 = self.QImage2Pixmap(5, divs)
                # self.div1.resize(QtCore.QSize(divs[0][0].shape[1], divs[0][0].shape[0]))
                Gray6 = Gray6.scaled(self.div6.width(), self.div6.height())
                self.div6.setPixmap(QtGui.QPixmap.fromImage(Gray6))

                Gray7 = self.QImage2Pixmap(6, divs)
                # self.div1.resize(QtCore.QSize(divs[0][0].shape[1], divs[0][0].shape[0]))
                Gray7 = Gray7.scaled(self.div7.width(), self.div7.height())
                self.div7.setPixmap(QtGui.QPixmap.fromImage(Gray7))