使用PyQt 来实现图片数据标注demo

PyQt参考学习的博客：https://blog.csdn.net/azuremouse/article/details/90338961

0.程序结构

 1.demo_pic 是保存了两张测试图片  苹果 和 橙子。 txt是程序运行并标注后生成的标签文件。

 

2. get_label_demo.ui 是使用Qtdesigner 实现的界面 get_label_demo.py 是其转换来的

3 . mian函数是实现具体功能的code部分

1.我实现的界面

导入文件夹按钮：选择一个包含图片的文件将其导入

中间区域用于显示图片。

这里假设有两个类别橙子和苹果。我们点击这两个按钮进行标注，标注之后自动跳转到下一张图片。

由于是demo功能没有那么全面，即没有上一张下一张那种功能等。

2.code部分

tip1: 使用  QFileDialog.getExistingDirectory 来实现导入文件夹

这里第一个参数原本设置为self 会出现报错。改为None

file_path =  QFileDialog.getExistingDirectory(None, "请选择模板保存路径...", "./")

tip2：

使用.clicked.connect 实现按钮触发事件 

这里ui 是我们的界面类的实例化对象

ui.pushButton_2 是指ui界面中的名为pushButton_2的按钮

这种界面按钮触发方式中的函数使用return 接收不到返回值

这里使用python中的partial 来实现参数的传递

partial(out_labela, ui, pic_path_list) 意为往函数out_labela中传入ui 和 pic_path_list 两个参数

ui.pushButton_2.clicked.connect(partial(out_labela, ui, pic_path_list))

3源码

mian.py

import sys
from PyQt5 import QtWidgets, QtCore, QtGui
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from functools import partial
import os
import get_label_demo
import cv2
import time
#获取文件夹
def openimage(ui):
    global index
    file_path =  QFileDialog.getExistingDirectory(None, "请选择模板保存路径...", "./")
    print(file_path)
    pic_list = os.listdir(file_path)
    print(pic_list)
    pic_path_list = []
    for i in pic_list:
        pic_path = file_path +'/'+ i
        # print(pic_path)
        pic_path_list.append((pic_path))
    print(pic_path_list)
    out_pic(ui,index, pic_path_list)

def out_pic(ui, index, pic_path_list):
    jpg = QtGui.QPixmap(pic_path_list[index]).scaled(ui.label.width(), ui.label.height())
    ui.label.setPixmap(jpg)
    ui.pushButton_2.clicked.connect(partial(out_labela, ui, pic_path_list))
    ui.pushButton_3.clicked.connect(partial(out_labelb, ui, pic_path_list))

def out_labela(ui, pic_path_list):
    #苹果
    global index

    #now pic
    now_pic_path = pic_path_list[index]
    print(now_pic_path)
    print(type(now_pic_path))
    now_txt_path = now_pic_path.replace("jpg","txt")
    print(now_txt_path)
    with open(now_txt_path, "w") as f:
        f.write("苹果")
        f.close()

    #判断是否为最后一个pic
    if index == len(pic_path_list):
        return

    #next pic
    index +=1
    out_pic(ui, index, pic_path_list)
    print('a')
def out_labelb(ui, pic_path_list):
    #橙子
    global index

    # now pic
    now_pic_path = pic_path_list[index]
    print(now_pic_path)
    print(type(now_pic_path))
    now_txt_path = now_pic_path.replace("jpg", "txt")
    print(now_txt_path)
    with open(now_txt_path, "w") as f:
        f.write("橙子")
        f.close()

    #判断是否为最后一个pic
    if index == len(pic_path_list):
        return

    #next pic
    index +=1
    out_pic(ui, index, pic_path_list)
    print('b')

if __name__ == '__main__':
    index = 0
    app = QApplication(sys.argv)
    MainWindow = QMainWindow()
    ui = get_label_demo.Ui_MainWindow()
    ui.setupUi(MainWindow)
    MainWindow.show()
    ui.pushButton.clicked.connect(partial(openimage, ui))

    sys.exit(app.exec_())

#steps
#step1 获取文件夹 get name list
#step2 遍历图片
#step3 显示图片
#step4 标注触发
#step5 保存label

get_label_demo.py

# -*- coding: utf-8 -*-

# Form implementation generated from reading ui file 'get_label_demo.ui'
#
# Created by: PyQt5 UI code generator 5.15.1
#
# WARNING: Any manual changes made to this file will be lost when pyuic5 is
# run again.  Do not edit this file unless you know what you are doing.


from PyQt5 import QtCore, QtGui, QtWidgets


class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(800, 600)
        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        self.verticalLayout = QtWidgets.QVBoxLayout(self.centralwidget)
        self.verticalLayout.setObjectName("verticalLayout")
        self.pushButton = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton.setObjectName("pushButton")
        self.verticalLayout.addWidget(self.pushButton)
        self.label = QtWidgets.QLabel(self.centralwidget)
        self.label.setObjectName("label")
        self.verticalLayout.addWidget(self.label)
        self.pushButton_3 = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton_3.setObjectName("pushButton_3")
        self.verticalLayout.addWidget(self.pushButton_3)
        self.pushButton_2 = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton_2.setObjectName("pushButton_2")
        self.verticalLayout.addWidget(self.pushButton_2)
        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtWidgets.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 23))
        self.menubar.setObjectName("menubar")
        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtWidgets.QStatusBar(MainWindow)
        self.statusbar.setObjectName("statusbar")
        MainWindow.setStatusBar(self.statusbar)

        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):
        _translate = QtCore.QCoreApplication.translate
        MainWindow.setWindowTitle(_translate("MainWindow", "get_label_demo"))
        self.pushButton.setText(_translate("MainWindow", "导入文件夹"))
        self.label.setText(_translate("MainWindow", "显示图片"))
        self.pushButton_3.setText(_translate("MainWindow", "橙子"))
        self.pushButton_2.setText(_translate("MainWindow", "苹果"))

---

修改：之前是定义一个继承object对象的类来实例化界面，查看了一些教程，这种方式有点冗余，修改不灵活。

现优化为：定义一个继承PyQt5.QWidget来实例化界面， 整体流程大部分都是通过编写代码实现的。

添加了 上一张和下一张功能。 和按钮触发颜色变换功能。

1运行程序

2点击导入文件夹 选择文件夹之后 选择显示图片

 3点击类别并进行标注 点击的按钮颜色会变成 浅绿色 其他按钮颜色会变成白色

 4点击下一张跳转到下一张 点击上一张回跳到上一张

 代码part：

其中index用于索引图片

id用于索引类别按钮

import sys
import os
from PyQt5 import QtWidgets, QtCore, QtGui
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from functools import partial
class picture(QWidget):
    def __init__(self, classlist):
        super(picture, self).__init__()

        self.resize(800, 600)
        self.setWindowTitle("标注图片")

        self.label = QLabel(self)
        self.label.setText("   显示图片")
        self.label.setFixedSize(600, 400)
        self.label.move(100, 150)

        self.label.setStyleSheet("QLabel{background:white;}"
                                 "QLabel{color:rgb(300,300,300,120);font-size:10px;font-weight:bold;font-family:宋体;}"
                                 )

        #index
        self.index = 0

        #导入文件夹
        self.pic_path_list = []
        btn = QPushButton(self)
        btn.setText("导入文件夹")
        btn.move(10, 30)
        btn.clicked.connect(self.opendir)
        # print('self pic path is', self.pic_path_list)


        self.classlist = classlist
        #创建标注类别按钮
        #get cfg class list
        #for 循环创建
        btn_list = []
        for i in range(len(self.classlist)):
            btn = QPushButton(self)
            btn.setText(self.classlist[i])
            btn.move(10+80*i, 60)
            btn_list.append(btn)

        for id in range(len(btn_list)):
            btn_list[id].clicked.connect(partial(self.writelabel,btn_list[id].text(), btn_list, id))


        #图片显示部分
        btn = QPushButton(self)
        btn.setText("显示图片")
        btn.move(10 , 90)
        btn.clicked.connect(self.showimage)

        btn = QPushButton(self)
        btn.setText("上一张")
        btn.move(10+80, 90)
        btn.clicked.connect(self.show_beforeimage)

        btn = QPushButton(self)
        btn.setText("下一张")
        btn.move(10+160, 90)
        btn.clicked.connect(self.show_nextimage)



    #导入一个文件夹的图片
    # 返回包含图片路径的list
    def opendir(self):
        file_path =  QFileDialog.getExistingDirectory(self, "请选择模板保存路径...", "./")
        print(file_path)
        pic_list = os.listdir(file_path)
        print(pic_list)
        for i in pic_list:
            pic_path = file_path + '/' + i
            # print(pic_path)
            self.pic_path_list.append((pic_path))
        print(self.pic_path_list)


    #用于显示图片
    #
    def showimage(self):
        print("ok")
        jpg = QtGui.QPixmap(self.pic_path_list[self.index]).scaled(self.label.width(), self.label.height())
        self.label.setPixmap(jpg)

    def show_nextimage(self):
        self.index += 1
        jpg = QtGui.QPixmap(self.pic_path_list[self.index]).scaled(self.label.width(), self.label.height())
        self.label.setPixmap(jpg)

    def show_beforeimage(self):
        self.index -= 1
        jpg = QtGui.QPixmap(self.pic_path_list[self.index]).scaled(self.label.width(), self.label.height())
        self.label.setPixmap(jpg)


    def writelabel(self, label, btn_list, id):
        print(label)
        now_txt_path = self.pic_path_list[self.index].replace("jpg", "txt")
        print(now_txt_path)
        with open(now_txt_path, "w") as f:
            f.write(label)
            f.close()
        if self.index == len(self.pic_path_list)-1:
            sys.exit()
        #触发按钮颜色变化
        for b_Button_id in range(len(btn_list)):
            if b_Button_id == id :
                btn_list[b_Button_id].setStyleSheet('background-color: #7FFFAA')
            else:
                btn_list[b_Button_id].setStyleSheet('background-color: #FFFFFF')



if __name__ == "__main__":
    list = ["苹果", "橙子"]
    app = QtWidgets.QApplication(sys.argv)
    my = picture(list)
    my.show()
    sys.exit(app.exec_())