我的程序是通过摄像头采集一帧图像，点击开始标记按钮后显示在按钮左侧的lable中（灰色框）。
我希望可以在lable中加载的一帧图像中通过点击鼠标左键开始绘制矩形，按住左键拖动时改变矩形的形状，松开左键后完成绘制。（和lableImg软件类似）。
请问如何实现上述功能呢？我目前找到的方法是使用QPainter来实现，但是在我的代码中该如何实现呢？万分感谢！
附上我的完整代码如下：
# -*- coding: utf-8 -*-
import sys
import cv2
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5.QtGui import *

# ~ class MyLabel(QLabel):
    # ~ def paintEvent(self, event):
        # ~ width = self.pos2[0]-self.pos1[0]
        # ~ height = self.pos2[1] - self.pos1[1]     
        # ~ qp = QPainter()
        # ~ qp.begin(self)           
        # ~ qp.drawRect(self.pos1[0], self.pos1[1], width, height)        
        # ~ qp.end()

    # ~ def mousePressEvent(self, event):
        # ~ self.pos1[0], self.pos1[1] = event.pos().x(), event.pos().y()
        # ~ print("clicked")

    # ~ def mouseReleaseEvent(self, event):
        # ~ self.pos2[0], self.pos2[1] = event.pos().x(), event.pos().y()
        # ~ print("released")
        # ~ self.update()

class Ui_train_window(QtWidgets.QWidget):
    def setupUi(self, train_window):
        train_window.setObjectName("train_window")
        train_window.resize(690, 600)
        train_window.setMaximumSize(QtCore.QSize(690, 600))
        self.horizontalLayoutWidget = QtWidgets.QWidget(train_window)
        self.horizontalLayoutWidget.setGeometry(QtCore.QRect(0, 10, 681, 80))
        self.horizontalLayoutWidget.setObjectName("horizontalLayoutWidget")
        self.horizontalLayout = QtWidgets.QHBoxLayout(self.horizontalLayoutWidget)
        self.horizontalLayout.setContentsMargins(0, 0, 0, 0)
        self.horizontalLayout.setObjectName("horizontalLayout")
        self.lab_enter_name = QtWidgets.QLabel(self.horizontalLayoutWidget)
        self.lab_enter_name.setObjectName("lab_enter_name")
        self.horizontalLayout.addWidget(self.lab_enter_name)
        self.le_username = QtWidgets.QLineEdit(self.horizontalLayoutWidget)
        self.le_username.setObjectName("le_username")
        self.horizontalLayout.addWidget(self.le_username)
        self.lab_enter_pw = QtWidgets.QLabel(self.horizontalLayoutWidget)
        self.lab_enter_pw.setObjectName("lab_enter_pw")
        self.horizontalLayout.addWidget(self.lab_enter_pw)
        self.le_userpw = QtWidgets.QLineEdit(self.horizontalLayoutWidget)
        self.le_userpw.setObjectName("le_userpw")
        self.le_userpw.setEchoMode(QLineEdit.Password)
        self.horizontalLayout.addWidget(self.le_userpw)
        self.label = QtWidgets.QLabel(self.horizontalLayoutWidget)
        self.label.setObjectName("label")
        self.horizontalLayout.addWidget(self.label)
        self.le_ipadr = QtWidgets.QLineEdit(self.horizontalLayoutWidget)
        self.le_ipadr.setObjectName("le_ipadr")
        self.horizontalLayout.addWidget(self.le_ipadr)
        self.btn_openIPcam = QtWidgets.QPushButton(self.horizontalLayoutWidget)
        self.btn_openIPcam.setMinimumSize(QtCore.QSize(50, 10))
        self.btn_openIPcam.setObjectName("btn_openIPcam")
        self.horizontalLayout.addWidget(self.btn_openIPcam)

        self.show = QtWidgets.QLabel(train_window)
        self.show.setGeometry(QtCore.QRect(15, 110, 531, 371))
        self.show.setObjectName("show")
        self.show.setStyleSheet(("border:2px solid lightgray"))

        self.verticalLayoutWidget = QtWidgets.QWidget(train_window)
        self.verticalLayoutWidget.setGeometry(QtCore.QRect(560, 100, 111, 381))
        self.verticalLayoutWidget.setObjectName("verticalLayoutWidget")
        self.verticalLayout = QtWidgets.QVBoxLayout(self.verticalLayoutWidget)
        self.verticalLayout.setContentsMargins(0, 0, 0, 0)
        self.verticalLayout.setObjectName("verticalLayout")
        self.btn_start = QtWidgets.QPushButton(self.verticalLayoutWidget)
        self.btn_start.setObjectName("btn_start")
        self.verticalLayout.addWidget(self.btn_start)
        self.combo_label = QtWidgets.QComboBox(self.verticalLayoutWidget)
        self.combo_label.setObjectName("combo_label")
        self.combo_label.addItem("")
        self.combo_label.addItem("")
        self.combo_label.addItem("")
        self.combo_label.addItem("")
        self.combo_label.addItem("")
        self.combo_label.addItem("")
        self.verticalLayout.addWidget(self.combo_label)
        self.btn_save = QtWidgets.QPushButton(self.verticalLayoutWidget)
        self.btn_save.setObjectName("btn_save")
        self.verticalLayout.addWidget(self.btn_save)
        self.btn_finish = QtWidgets.QPushButton(self.verticalLayoutWidget)
        self.btn_finish.setObjectName("btn_finish")
        self.verticalLayout.addWidget(self.btn_finish)
        self.horizontalLayoutWidget_2 = QtWidgets.QWidget(train_window)
        self.horizontalLayoutWidget_2.setGeometry(QtCore.QRect(0, 500, 681, 71))
        self.horizontalLayoutWidget_2.setObjectName("horizontalLayoutWidget_2")
        self.horizontalLayout_2 = QtWidgets.QHBoxLayout(self.horizontalLayoutWidget_2)
        self.horizontalLayout_2.setContentsMargins(0, 0, 0, 0)
        self.horizontalLayout_2.setObjectName("horizontalLayout_2")
        self.lab_stepnum = QtWidgets.QLabel(self.horizontalLayoutWidget_2)
        self.lab_stepnum.setObjectName("lab_stepnum")
        self.horizontalLayout_2.addWidget(self.lab_stepnum)
        self.le_stepnm = QtWidgets.QLineEdit(self.horizontalLayoutWidget_2)
        self.le_stepnm.setObjectName("le_stepnm")
        self.horizontalLayout_2.addWidget(self.le_stepnm)
        self.lab_batchsize = QtWidgets.QLabel(self.horizontalLayoutWidget_2)
        self.lab_batchsize.setObjectName("lab_batchsize")
        self.horizontalLayout_2.addWidget(self.lab_batchsize)
        self.le_batchsize = QtWidgets.QLineEdit(self.horizontalLayoutWidget_2)
        self.le_batchsize.setObjectName("le_batchsize")
        self.horizontalLayout_2.addWidget(self.le_batchsize)
        self.btn_pretrain = QtWidgets.QPushButton(self.horizontalLayoutWidget_2)
        self.btn_pretrain.setObjectName("btn_pretrain")
        self.horizontalLayout_2.addWidget(self.btn_pretrain)
        self.btn_retrain = QtWidgets.QPushButton(self.horizontalLayoutWidget_2)
        self.btn_retrain.setObjectName("btn_retrain")
        self.horizontalLayout_2.addWidget(self.btn_retrain)
        self.show.raise_()
        self.horizontalLayoutWidget.raise_()
        self.verticalLayoutWidget.raise_()
        self.horizontalLayoutWidget_2.raise_()

        self.retranslateUi(train_window)
        self.btn_openIPcam.clicked.connect(self.IPinfo)
        self.btn_start.clicked.connect(self.labledata)
        QtCore.QMetaObject.connectSlotsByName(train_window)

    def retranslateUi(self, train_window):
        _translate = QtCore.QCoreApplication.translate
        train_window.setWindowTitle(_translate("train_window", "训练工具"))
        self.lab_enter_name.setText(_translate("train_window", "输入IP摄像头用户名："))
        self.lab_enter_pw.setText(_translate("train_window", "输入IP摄像头密码："))
        self.label.setText(_translate("train_window", "IP地址"))
        self.btn_openIPcam.setText(_translate("train_window", "连接IP摄像头"))
        self.show.setText(_translate("train_window", ""))
        self.btn_start.setText(_translate("train_window", "开始标记"))
        self.combo_label.setItemText(0, _translate("train_window", "a"))
        self.combo_label.setItemText(1, _translate("train_window", "b"))
        self.combo_label.setItemText(2, _translate("train_window", "c"))
        self.combo_label.setItemText(3, _translate("train_window", "d"))
        self.combo_label.setItemText(4, _translate("train_window", "e"))
        self.combo_label.setItemText(5, _translate("train_window", "f"))
        self.btn_save.setText(_translate("train_window", "保存"))
        self.btn_finish.setText(_translate("train_window", "标记结束"))
        self.lab_stepnum.setText(_translate("train_window", "迭代次数："))
        self.lab_batchsize.setText(_translate("train_window", "batchsize："))
        self.btn_pretrain.setText(_translate("train_window", "预训练"))
        self.btn_retrain.setText(_translate("train_window", "重新训练"))

    def IPinfo(self):
        username = self.le_username.text()
        password = self.le_userpw.text()
        ipaddress = self.le_ipadr.text()
        #cam_rtsp_addr = "rtsp://" + username + ":" + password + "@" + ipaddress + "/h264/ch33/main/av_stream"
        #self.camcapture = cv2.VideoCapture(cam_rtsp_addr)
        self.camcapture = cv2.VideoCapture(0)
        self.timer = QtCore.QTimer()
        self.timer.start()
        self.timer.setInterval(0.3) 
        self.timer.timeout.connect(self.camshow)

    def camshow(self):
        global showImage
        _ , camimg = self.camcapture.read()
        camimg = cv2.cvtColor(camimg, cv2.COLOR_BGR2RGB)
        showImage = QtGui.QImage(camimg.data, camimg.shape[1], camimg.shape[0], QtGui.QImage.Format_RGB888)

    def labledata(self):
        self.show.setPixmap(QtGui.QPixmap.fromImage(showImage))

if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)
    Window = QtWidgets.QWidget()
    ui = Ui_train_window()
    ui.setupUi(Window)
    Window.show()
    sys.exit(app.exec_())
使用labelImg对图片做标注，标注的的结果是怎么使用的？

justinlonger

1个回复
