本文实例为大家分享了pyqt5实现简单数据标注工具的具体代码，分类用，供大家参考，具体内容如下
第一个最大的图片是当前要标注的类别，接下来的两个图片是对接下来会出现的图片的预览（方便连续点好几个）。分类之后的会保存到和按钮名字一样的文件夹里面，如果文件夹不存在就会自动新建一个（makedirs）。如果中断了标注，可以修改代码中的self.idx属性，从某个位置开始。
视频效果：地址链接
接下来是代码：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
from pyqt5.qtwidgets import qapplication,qpushbutton,qlabel,qmainwindow
from pyqt5.qtgui import qfont,qpixmap
import sys,os
import shutil
def copyfile(srcfile, dstfile):#用来复制文件，源文件会保留
if not os.path.isfile(srcfile):
print("%s not exist!" % srcfile)
else:
f_path, f_name = os.path.split(dstfile) # 分离文件名和路径
if not os.path.exists(f_path):
os.makedirs(f_path) # 创建路径
shutil.copyfile(srcfile, dstfile) # 复制文件
print("copy %s -> %s" % (srcfile, dstfile))
class mainform(qmainwindow):
def __init__(self):
super(mainform, self).__init__()
self.img_path="faces/" #文件夹和py文件要再同一个目录下面
self.img_list=os.listdir(self.img_path) #获取目录下的所有文件
self.idx=0#可以改这里，选择程序运行的时候第一个显示的图片是哪一个
self.initui()
self.show()
def initui(self):
font=qfont()
font.setpixelsize(20)#新建一个字体控件
self.setwindowtitle("label_me")#设置窗体的标题
self.setgeometry(0,0,900,600)#位置和大小
button_list=["chandler","phoebe","joey","monica","rachel","ross","others","thing",]#这里是显示的按钮们，也是可能的类别数
for idx, label_name in enumerate(button_list):
button=qpushbutton(label_name,self)
button.move(idx*110+20,500)
button.setfont(font)
button.setfixedheight(35)
button.clicked.connect(self.classify)#动态控件绑定同一个事件，根据事件的sender判断是哪个按钮按下
self.lbl_list=[]#存放显示图片的label 的list
for i in range(self.get_remainder()):
self.pix = qpixmap(self.img_path+self.img_list[self.idx+i])
label_img = qlabel(self)
label_img.setgeometry(360*i+10, 400-100*(3-i)+40, 100*(3-i)+40,100*(3-i)+40)
label_img.setstylesheet("border:2px solid red")
label_img.setpixmap(self.pix)#设置label控件要显示的图片
label_img.setscaledcontents(true)
self.lbl_list.append(label_img)
def get_remainder(self):#打算是要显示3个label图片，如果是到了最后，显示不了那么多了。
r=len(self.img_list)-self.idx
if r>3:
r=3
return r
def clear_lbls(self):#最后的时候会用到，不显示某些label
for i in range(len(self.lbl_list)):
self.lbl_list[i].hide()
def classify(self):
sender = self.sender()
dir_path=sender.text()+"/"#获取按钮的text属性
current_img_path=self.img_list[self.idx]#获取刚刚被分类的图片的路径
copyfile(self.img_path+current_img_path , dir_path+current_img_path)
self.idx +=1#下一个图片
img_full_path = [self.img_path + self.img_list[self.idx + i] for i in range(self.get_remainder())]
self.clear_lbls()
for i in range(self.get_remainder()):
pix = qpixmap(img_full_path[i])
self.lbl_list[i].setpixmap(pix)
self.lbl_list[i].show()
self.setwindowtitle("当前是第 %d 个图片"%self.idx)
app=qapplication(sys.argv)
f=mainform()
sys.exit(app.exec())
|
如果有什么需要完善的地方，请提出。
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。
原文链接：https://blog.csdn.net/york1996/article/details/84946374