PyQt5实现简单数据标注工具
作者:York1996 发布时间:2021-10-24 04:55:54
标签:PyQt5,标注工具
本文实例为大家分享了PyQt5实现简单数据标注工具的具体代码,分类用,供大家参考,具体内容如下
第一个最大的图片是当前要标注的类别,接下来的两个图片是对接下来会出现的图片的预览(方便连续点好几个)。分类之后的会保存到和按钮名字一样的文件夹里面,如果文件夹不存在就会自动新建一个(makedirs)。如果中断了标注,可以修改代码中的self.idx属性,从某个位置开始。
视频效果:地址链接
接下来是代码:
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


猜你喜欢
- 前言kylin是一个开源的OLAP分析引擎,具有亚秒级查询大表的能力通过kylin提供的cube预构建功能,省去了不断写sql查询hive的
- 很久以前就知道微软的Petshop的很经典,昨天抽出时间去学习,一开始还真的不适应,什么成员资格,还真的看不太懂,运行petshop想从登陆
- 环境:Python3.7依赖库:import datetimeimport randomimport requestsimport hash
- 如下所示:from splinter.browser import Browserb = Browser('chrome')
- 本文的目标是将表1论文及出版物与表2出版物及其指标进行匹配。利用pandas的merge函数实现表的左外连接。左外连接即左表的外连接,左表保
- 前言通常我们在一个站站点进行采集的时候,如果是小站的话 我们使用scrapy本身就可以满足。但是如果在面对一些比较大型的站点的时候,单个sc
- collections中defaultdict的用法一、字典的键映射多个值将下面的列表转换成字典一个字典就是一个键对应一个单值得映射,而上面
- 1.打开文件:f=open(r'E:\PythonProjects\test7\a.txt',mode='rt
- 本文实例讲述了python去除文件中空格、Tab及回车的方法。分享给大家供大家参考,具体如下:在最近的开发工作中,为了应付比赛赶进度,服务端
- 功能性的文章直接用几个最简单的实现表达:xlsxwriter库的核心就是其Workbook对象。创建一个指定名字的xlsx文件:import
- Python等工具确实是不错的工具,但是有时候不管是基础的Python还是Python的软件包都让我觉得对中文不是很亲近。时不时地遇到一点问
- python tkinter按钮Button的使用创建和设置窗口from tkinter import *#创建窗口对象root = Tk(
- 这里是一个基于GMap2和XML的小例子,数据存在XML文件中 ,这是最简单的模式,却相当地有用。实例的网址是: http://sunjia
- 实现的思路:将准备好的图片通过opencv读取出来,并将其设置好帧数等参数后合成为无声视频。最后通过moviepy编辑视频将背景音乐加入到视
- 本文介绍了django反向解析URL和URL命名空间,分享给大家,具体如下:首先明确几个概念:1.在html页面上的内容特别是向用户展示的u
- 众所周知,python是写爬虫的利器,今天作者用python写一个小爬虫爬下一个段子网站的众多段子。目标段子网站为“http://ishuo
- 这篇文章主要介绍了python如何基于redis实现ip代理池,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,
- 本文给大家介绍如何判断表单验证的实例代码,在没给大家介绍正文之前,先给大家介绍下插件。插件介绍先上一个图:下载地址:https://gith
- 如果将程序员分为本文的8种类型,你会是哪一种呢?在求职的时候,相信很多人都被问过这样的问题,“你对自己未来5年的职业规划是怎么样的?” 每当
- 每次讲解SQL Server里的锁和阻塞(Locking & Blocking)都会碰到的问题:在SQL Server里,为什么我们