【QT】利用pyqt5实现简单界面
Topic: 利用pyqt5编写简单界面
Env：win10 + Pycharm2018 + Python 3.6.8
Date: 2019/4/29 by hw_Chen2018 CSDN:https://blog.csdn.net/qq_34198088/article/details/97893081
基本功能：利用摄像头采集使用者头像，点击“采集照片”，当前图像会“暂存”并显示在右边，“暂存”数量3。然后依次填入姓名，选择姓名，最后点击“保存”，保存照片在固定路径下。如果“暂存”图像张数小于3、未填入姓名、选择性别，会提示未进行相应操作。选择“清空”，会清空当前“暂存”图像和输入姓名字符串，以便继续录入。（ 因为我这里使用的双目摄像头，显示的左右相机的输出，如果接入正常单目USB摄像头，则是正常的；与脚本同一级目录下有个img文件夹。功能简单，调用第三方库少，转载请注明出处！
界面如下图：
from cap_info import * import cv2 as cv from PyQt5 import QtCore, QtGui, QtWidgets import sys import matplotlib matplotlib.rcParams['font.sans-serif'] = ['SimHei'] matplotlib.rcParams['font.family'] = 'sans-serif' matplotlib.rcParams['axes.unicode_minus'] = False if __name__ == "__main__": app = QtWidgets.QApplication(sys.argv) w = mywindow() w.photo_pb.clicked.connect(w.display_img) w.show() sys.exit(app.exec_())
脚本can_info.py如下：
# -*- coding: utf-8 -*- # Form implementation generated from reading ui file 'cap_info.ui' # # Created by: PyQt5 UI code generator 5.11.3 # # WARNING! All changes made in this file will be lost! from PyQt5 import QtCore, QtGui, QtWidgets import cv2 as cv import sys import os class Ui_Form(object): def setupUi(self, Form): Form.setObjectName("Form") Form.resize(480, 480) Form.setMinimumSize(QtCore.QSize(480, 480)) Form.setMaximumSize(QtCore.QSize(480, 480)) self.photo_pb = QtWidgets.QPushButton(Form) self.photo_pb.setGeometry(QtCore.QRect(190, 290, 81, 31)) self.photo_pb.setStyleSheet("font: 14pt \"华文行楷\";") self.photo_pb.setObjectName("photo_pb") self.save_pb = QtWidgets.QPushButton(Form) self.save_pb.setGeometry(QtCore.QRect(150, 440, 75, 31)) self.save_pb.setStyleSheet("font: 14pt \"华文行楷\";") self.save_pb.setObjectName("save_pb") self.photo_lbl = QtWidgets.QLabel(Form) self.photo_lbl.setGeometry(QtCore.QRect(70, 30, 321, 251)) self.photo_lbl.setMinimumSize(QtCore.QSize(321, 251)) self.photo_lbl.setMaximumSize(QtCore.QSize(321, 251)) self.photo_lbl.setFrameShape(QtWidgets.QFrame.Box) self.photo_lbl.setText("") self.photo_lbl.setObjectName("photo_lbl") self.label = QtWidgets.QLabel(Form) self.label.setGeometry(QtCore.QRect(120, 350, 51, 20)) self.label.setStyleSheet("font: 14pt \"华文行楷\";") self.label.setObjectName("label") self.label_2 = QtWidgets.QLabel(Form) self.label_2.setGeometry(QtCore.QRect(120, 390, 51, 20)) self.label_2.setStyleSheet("font: 14pt \"华文行楷\";") self.label_2.setObjectName("label_2") self.name = QtWidgets.QLineEdit(Form) self.name.setGeometry(QtCore.QRect(180, 350, 113, 20)) self.name.setObjectName("name") self.radioButton = QtWidgets.QRadioButton(Form) self.radioButton.setGeometry(QtCore.QRect(180, 390, 41, 16)) self.radioButton.setObjectName("radioButton") self.radioButton_2 = QtWidgets.QRadioButton(Form) self.radioButton_2.setGeometry(QtCore.QRect(240, 390, 41, 16)) self.radioButton_2.setObjectName("radioButton_2") self.photo1 = QtWidgets.QLabel(Form) self.photo1.setGeometry(QtCore.QRect(410, 30, 51, 71)) self.photo1.setMinimumSize(QtCore.QSize(51, 71)) self.photo1.setMaximumSize(QtCore.QSize(51, 71)) self.photo1.setFrameShape(QtWidgets.QFrame.Box) self.photo1.setObjectName("photo1") self.photo2 = QtWidgets.QLabel(Form) self.photo2.setGeometry(QtCore.QRect(410, 120, 51, 71)) self.photo2.setMinimumSize(QtCore.QSize(51, 71)) self.photo2.setMaximumSize(QtCore.QSize(51, 71)) self.photo2.setFrameShape(QtWidgets.QFrame.Box) self.photo2.setObjectName("photo2") self.photo3 = QtWidgets.QLabel(Form) self.photo3.setGeometry(QtCore.QRect(410, 210, 51, 71)) self.photo3.setMinimumSize(QtCore.QSize(51, 71)) self.photo3.setMaximumSize(QtCore.QSize(51, 71)) self.photo3.setFrameShape(QtWidgets.QFrame.Box) self.photo3.setObjectName("photo3") self.clear_pb = QtWidgets.QPushButton(Form) self.clear_pb.setGeometry(QtCore.QRect(250, 440, 71, 31)) self.clear_pb.setStyleSheet("font: 14pt \"华文行楷\";") self.clear_pb.setObjectName("clear_pb") self.label_3 = QtWidgets.QLabel(Form) self.label_3.setGeometry(QtCore.QRect(365, 300, 101, 21)) self.label_3.setStyleSheet("font: 75 9pt \"Adobe Arabic\";") self.label_3.setObjectName("label_3") self.photo_num = QtWidgets.QLabel(Form) self.photo_num.setGeometry(QtCore.QRect(430, 300, 21, 20)) self.photo_num.setText("") self.photo_num.setObjectName("photo_num") self.retranslateUi(Form) QtCore.QMetaObject.connectSlotsByName(Form) def retranslateUi(self, Form): _translate = QtCore.QCoreApplication.translate Form.setWindowTitle(_translate("Form", "采集界面")) self.photo_pb.setText(_translate("Form", "采集照片")) self.save_pb.setText(_translate("Form", "保存")) self.label.setText(_translate("Form", "姓名：")) self.label_2.setText(_translate("Form", "性别：")) self.radioButton.setText(_translate("Form", "男")) self.radioButton_2.setText(_translate("Form", "女")) self.photo1.setText(_translate("Form", "照片1")) self.photo2.setText(_translate("Form", "照片2")) self.photo3.setText(_translate("Form", "照片3")) self.clear_pb.setText(_translate("Form", "清空")) self.label_3.setText(_translate("Form", "照片数量：")) class mywindow(Ui_Form, QtWidgets.QWidget): def __init__(self): super(mywindow, self).__init__() self.setupUi(self) self.cap = cv.VideoCapture(0) if not self.cap.isOpened(): QtWidgets.QMessageBox.warning(self, "警告", "未成功打开摄像头！") self.video_timer = QtCore.QTimer() self.photo_pb.clicked.connect(self.video_play_slot) # 显示每一帧图像 self.captureImg_flag = False # 截图控制符 self.captureImg_count = 0 # 截图数目 self.img_list = [] # 保存截图 self.captureImg_num = 0 # 截图数量 self.save_pb.clicked.connect(self.save_info) # 保存信息 self.clear_pb.clicked.connect(self.clear_content) def video_play_slot(self): if not self.cap.isOpened(): QtWidgets.QMessageBox.warning(self, "警告", "未成功打开摄像头！请关闭界面并重新打开！") else: if not self.video_timer.isActive(): self.video_timer.start(30) self.video_timer.timeout.connect(self.display_img) else: self.captureImg_flag = True def closeEvent(self, QCloseEvent): ''' 窗口关闭事件 :param QCloseEvent: :return: ''' reply = QtWidgets.QMessageBox.question(self, "确认", "确认退出吗?", QtWidgets.QMessageBox.Yes|QtWidgets.QMessageBox.No) if reply == QtWidgets.QMessageBox.Yes: self.video_timer.stop() self.cap.release() QCloseEvent.accept() else: QCloseEvent.ignore() def display_img(self): ''' 显示每一帧图像 :return: 无 ''' self.photo_num.setText(str(len(self.img_list))) # 显示当前采集的照片数量 ret, frame = self.cap.read() if not ret: print("cannot receive this frame.") else: frame = cv.cvtColor(frame, cv.COLOR_BGR2RGB) height, width, bytesPerComponent = frame.shape bytesPerLine = bytesPerComponent * width q_image = QtGui.QImage(frame.data, width, height, bytesPerLine, QtGui.QImage.Format_RGB888).scaled(self.photo_lbl.width(), self.photo_lbl.height()) self.photo_lbl.setPixmap(QtGui.QPixmap.fromImage(q_image)) if self.captureImg_flag: # if self.captureImg_count == 1: # QtWidgets.QMessageBox.information(self, "提醒", "请尽量保持所采集的照片人脸姿态多样性！") if len(self.img_list) >= 3: self.img_list.pop(0) frame1 = frame.copy() temp = self.captureImg_count % 3 height, width, bytesPerComponent = frame1.shape bytesPerLine = bytesPerComponent * width q_image = QtGui.QImage(frame1.data, width, height, bytesPerLine,QtGui.QImage.Format_RGB888).scaled(self.photo1.width(), self.photo1.height()) if temp == 0: self.photo1.setPixmap(QtGui.QPixmap.fromImage(q_image)) self.img_list.append(frame1) # print(len(self.img_list)) elif temp == 1: self.photo2.setPixmap(QtGui.QPixmap.fromImage(q_image)) self.img_list.append(frame1) # print(len(self.img_list)) elif temp == 2: self.photo3.setPixmap(QtGui.QPixmap.fromImage(q_image)) self.img_list.append(frame1) # print(len(self.img_list)) self.captureImg_flag = False self.captureImg_count += 1 def save_info(self): flag1 = False flag2 = False flag3 = False if len(self.img_list) < 3: # 判断照片数量 QtWidgets.QMessageBox.warning(self, "警告", "照片数量小于三张！") else: flag1 = True name_str = self.name.text() if len(name_str) == 0: # 判断是否输入姓名 QtWidgets.QMessageBox.warning(self, "警告", "姓名为空！") else: flag2 = True if self.radioButton.isChecked() == self.radioButton_2.isChecked(): # 判断是否选择性别 QtWidgets.QMessageBox.warning(self, "警告", "未选择性别！") else: flag3 = True if flag2 and flag1 and flag3: dirname, filename = os.path.split((os.path.abspath(sys.argv[0]))) # 获取当前主运行文件绝对地址 targetpath = dirname + os.path.sep + 'img' # 构造文件路径 if not os.path.exists(targetpath): # 判断路径是否存在，不存在则创建 os.makedirs(targetpath) # os.chdir(targetpath) # 设置当前工作路径 temp_info_text = open(dirname + os.path.sep + 'img' + os.path.sep + 'info.txt', 'x+') temp_info_text.close() os.chdir(targetpath) info_text = open('info.txt','a') for _num in range(3): imgname = name_str + '_' + str(_num) + '.jpg' tempimg = cv.cvtColor(self.img_list[(len(self.img_list) - _num - 1)],cv.COLOR_RGB2BGR) cv.imencode('.jpg', tempimg)[1].tofile(imgname) temppath = targetpath + os.path.sep + imgname + '#' info_text.write(temppath) # 保存照片路径 if self.radioButton.isChecked(): gender = '男' + '#' elif self.radioButton_2.isChecked(): gender = '女' + '#' info_text.write(gender) # 写入性别信息 info_text.write('\n') info_text.close() QtWidgets.QMessageBox.about(self, "保存", "保存成功！") def clear_content(self): reply = QtWidgets.QMessageBox.question(self, "清空", "确认清空所有吗?", QtWidgets.QMessageBox.Yes|QtWidgets.QMessageBox.No) if reply == QtWidgets.QMessageBox.Yes: self.photo1.clear() self.photo2.clear() self.photo3.clear() self.img_list.clear() self.name.clear() self.captureImg_count = 0 # self.radioButton.setChecked(False) # self.radioButton_2.setChecked(False)
