时间过得真快，最近好久没有和大家闲扯淡了，好像，好像你们啊！！最近工作比较忙，一直想写点东西，但是身不由己啊。好不容易抽出一点时间写了一个工具——图像标注工具，这个工具与github上的labelImg很像，坦白承认我的却抄袭labelImg的部分标注方法（臭不要脸的抄袭，赶快滚，有多远滚多远）。
（2018年8月14日）最近发现一个比较好用的标注工具labelme，我个人感觉这个应该比labelImg好用，这里是labelme的github的连接，关于如何使用这个工具请移步这里。
我这这一个工具的目的：
1.最近在学习python，所以想练练手；
2.labelImg只能够标注矩形目标区域，局限性比较多，我想写一个更加通用的工具（更加通用可能有点太自大了，写到最好发现，我写的东西也有很多局限性，后面我会不断的迭代）；
现在给大家展示一下这个软件的界面：
这个界面是初始界面，下面的界面是工作时的界面：
散点标注为下图：
这里我主要使用python中的tkinter库实现界面（我在网上看到：有人说这个库不好，至于这个库的好坏我不过评论，仁者见仁智者见智吧），标注文件目前仅支持XML。
目前存在的问题：
1.未知的bug（自己测试难免会有问题）
2.离散点标注，目前要求标注四个点；
3.标注文件保存格式仅支持XML；
4.代码书写不规范；
5.仅支持中文；
6.部分弹窗不友好。
针对以上问题，会在后期的版本改正这些问题，改正完2，4，6后，我会将这个工程共享到github。众人拾柴火焰高，求大神帮忙，一起完善改进，谢谢！！
修改于2018年8月14日，主要优化了界面和代码，优化后的界面如下图：
下面贴出一部分代码：
#-*- coding:utf-8 -*-
import codecs
from lxml import etree
from xml.etree import ElementTree
from xml.etree.ElementTree import Element, SubElement
class XMLWrite:
def __init__(self,folder_path,file_name):
self.folder_path = folder_path
self.file_name = file_name
self.top = Element('Images')
self.object = None
def create_object(self,path,object_name = 'objectName',tag_name = 'FilePath'):
self.object = SubElement(self.top, object_name)
self.object.set(tag_name, path)
def set_lable_info(self, lable ,lable_name='LableName'):
if self.object == None:
raise Exception('没有创建object对象，create_object方法')
plate = SubElement(self.object, lable_name)
plate.text = lable
def set_object_info(self,tag_name = 'Size',**args):
if self.object == None:
raise Exception('没有创建object对象，create_object方法')
size = SubElement(self.object, tag_name)
for key in args:
sub = SubElement(size, key)
sub.text = str(args[key])
def set_position_rect(self,tag_name = 'Position',**args):
if self.object == None:
raise Exception('没有创建object对象，create_object方法')
point = SubElement(self.object, tag_name)
for key in args:
sub = SubElement(point, key)
sub.text = str(args[key])
def set_position_scat(self, tag_name = 'Position',info = None):
if self.object == None:
raise Exception('没有创建object对象，create_object方法')
point = SubElement(self.object,tag_name)
for key in info:
sub = SubElement(point, key)
sub.set('x',str(info[key][0]))
sub.set('y', str(info[key][1]))
def prettify(self, elem):
rough_string = ElementTree.tostring(elem, 'utf8')
root = etree.fromstring(rough_string)
return etree.tostring(root, pretty_print=True, encoding='utf-8').replace(" ".encode(), "\t".encode())
def save(self,targetFile=None):
#root = self.genXML()
out_file = None
if targetFile is None:
out_file = codecs.open(self.file_name + '.xml', 'w', encoding='utf-8')
else:
out_file = codecs.open(targetFile, 'w', encoding='utf-8')
prettifyResult = self.prettify(self.top)
out_file.write(prettifyResult.decode('utf8'))
out_file.close()
这里更新有添加了JSON格式，JSON格式很简单，大家在使用过程中直接把XML改成JSON就可以了，这一块我后期会完善，下面是JSON格式的代码：
#-*- coding:utf-8 -*-
import json
class JSON:
def __init__(self):
pass
def JSONRead(self,path):
with open(path, 'r') as json_f:
data = json.load(json_f)
return data
def JSONWrite(self,path,dict_info):
with open(path,'w') as json_f:
json.dump(dict_info, json_f)上面主要用于处理xml文件的。
#!/usr/bin/env python3
# coding:utf-8
import os
import filetype
import tkinter as tk
from lib import XML_LIB
import tkinter.ttk as ttk
from tkinter import simpledialog
from PIL import Image, ImageTk
import tkinter.messagebox as mb
import tkinter.filedialog as dir
'''
这是一个基于tkinter库的图像标注工具，目前支持标注矩形区域和散点区域，标注文件支持XML和JSON格式
'''
IMAGE_FILE_TYPE = ['jpg','JPG','PNG','png','bmp','BMP']
class MainGUI(tk.Frame):
'''继承tkinker中的frame类，实现可以自动化调整串口的大小'''
img = None
def __init__(self,root):
super().__init__(root)
self.root = root
self.image_index = 0 #当前处理的图像索引
self.image_paths = [] #工作目录下的所有文件路径
self.image_names = []
self.image_label = []
self.work_dir = ''
self.factor = 1.0 #图像缩放因子
self.x = 0;self.y = 0 #存储坐标
self.clicked = False
self.number = 0 #用于矩形画图时直线提醒
self.lable_Type = 0 #标注类型
self.scatter_point = []
self.lable_save_file = XML_LIB.XMLWrite('.','test')
self.initComponent(root)
#self.initMenu(root)
def initComponent(self,root):
'''初始化界面上的控件'''
root.rowconfigure(0,weight=1)
root.columnconfigure(0,weight=1)
#self.initMenu(root) # 为顶级窗体添加菜单项
# 设置grid布局位置
self.grid(row=0, column=0, sticky=tk.NSEW)
# 设置行列权重，保证内建子组件会拉伸填充
self.rowconfigure(0, weight=1);
self.columnconfigure(0, weight=1)
#水平方向推拉组件
self.panedwin = ttk.PanedWindow(self,orient=tk.HORIZONTAL)
self.panedwin.grid(row=0, column=0, sticky=tk.NSEW)
self.frame_left = ttk.Frame(self.panedwin, relief=tk.SUNKEN)
self.frame_left.grid(row=0, column=0, sticky=tk.NS)
self.panedwin.add(self.frame_left, weight=1)
self.initPlayList()
self.frame_right = ttk.Frame(self.panedwin,relief=tk.SUNKEN)
self.frame_right.grid(row=0,column=0,sticky=tk.NSEW)
self.frame_right.columnconfigure(0,weight=1)
# self.frame_right.rowconfigure(0,weight=1)
# self.frame_right.rowconfigure(1,weight=9)
self.panedwin.add(self.frame_right,weight=50)
lable = tk.Label(self.frame_right,font='Times 9 bold',text="图片展示",fg='#4876FF',anchor='nw')
lable.grid(row=1, column=0,padx=6,pady=6,sticky=tk.NSEW)
canvas_frame = ttk.Frame(self.frame_right)
canvas_frame.grid(row=2, column=0)
self.canvas = tk.Canvas(canvas_frame,background='white')
self.canvas.grid(row=0, column=0,sticky=tk.NSEW)
self.canvas.bind("<Motion>", self.mouse_on_canvas)#self.canvas.bind("<B1-Motion>", self.mouse_move)
self.canvas.bind('<Button-1>',self.left_mouse_click)
#self.canvas.bind('<Button-1>',self.left_mouse_click_press)
#self.canvas.bind('<ButtonRelease-1>', self.left_mouse_click_release)
self.showImage('logo.png')
# 右侧Frame帧第二行添加控制按钮
self.frm_control = ttk.Frame(self.frame_right, relief=tk.RAISED) # 四个方向拉伸
self.frm_control.grid(row=0, column=0, sticky=tk.NSEW)
self.initCtrl() # 添加滑块及按钮
def initMenu(self, master):
'''初始化菜单'''
mbar = tk.Menu(master) # 定义顶级菜单实例
fmenu = tk.Menu(mbar, tearoff=False) # 在顶级菜单下创建菜单项
mbar.add_cascade(label=' 文件 ', menu=fmenu, font=('Times', 20, 'bold')) # 添加子菜单
fmenu.add_command(label="打开", command=self.menu_click_event)
fmenu.add_command(label="保存", command=self.menu_click_event)
fmenu.add_separator() # 添加分割线
fmenu.add_command(label="退出", command=master.quit())
etmenu = tk.Menu(mbar, tearoff=False)
mbar.add_cascade(label=' 编辑 ', menu=etmenu)
for each in ['复制', '剪切', '合并']:
etmenu.add_command(label=each, command=self.menu_click_event)
master.config(menu=mbar) # 将顶级菜单注册到窗体
def initCtrl(self):
'''初始化控制滑块及按钮'''
frm_but = ttk.Frame(self.frm_control, padding=2) # 控制区第二行放置按钮及标签
frm_but.grid(row=0, column=0, sticky=tk.EW)
self.initButton()
def initPlayList(self):
'''初始化树状视图'''
self.frame_left.rowconfigure(0, weight=1) # 左侧Frame帧行列权重配置以便子元素填充布局
self.frame_left.columnconfigure(0, weight=1) # 左侧Frame帧中添加树状视图
self.frame_left.rowconfigure(1, weight=1) # 左侧Frame帧行列权重配置以便子元素填充布局
self.frame_left.columnconfigure(1, weight=1) # 左侧Frame帧中添加树状视图
#文件名称列表
self.image_name_tree = ttk.Treeview(self.frame_left, selectmode='browse', show='tree', padding=[0, 0, 0, 0])
self.image_name_tree.grid(row=0, column=0, sticky=tk.NSEW) # 树状视图填充左侧Frame帧
self.image_name_tree.column('#0', width=150) # 设置图标列的宽度，视图的宽度由所有列的宽决定
# 一级节点parent='',index=第几个节点,iid=None则自动生成并返回，text为图标右侧显示文字
# values值与columns给定的值对应
#标签名称列表
self.lable_tree = ttk.Treeview(self.frame_left, selectmode='browse', show='tree', padding=[0, 0, 0, 0])
self.lable_tree.grid(row=1, column=0, sticky=tk.NSEW) # 树状视图填充左侧Frame帧
self.lable_tree.column('#0', width=150) # 设置图标列的宽度，视图的宽度由所有列的宽决定
# 一级节点parent='',index=第几个节点,iid=None则自动生成并返回，text为图标右侧显示文字
# values值与columns给定的值对应
self.create_lable_tree(self.lable_tree)
def menu_click_event(self):
'''菜单事件'''
pass
def create_image_tree(self,tree,work_path):
tr_root = tree.insert("", 0, None, open=True, text='工作目录',values=('root')) # 树视图添加根节点
node = tree.insert(tr_root, 0, None, open=True, text=work_path,values=('path')) # 根节点下添加一级节点
for i in range(len(self.image_names)):
tree.insert(node, 0, None, text=self.image_names[i],values=(self.image_names[i])) # 添加二级节点
tree.bind('<ButtonRelease-1>', self.image_tree_click) # 绑定单击离开事件
#树形条目点击事件
def image_tree_click(self,event): # 单击
for item in self.image_name_tree.selection():
item_text = self.image_name_tree.item(item, "values")
if item_text[0] in ['root','path']:
continue
image_path = self.work_dir + '/' + item_text[0]
self.showImage(image_path)
#创建标签树
def create_lable_tree(self,tree):
self.tr_root = tree.insert("", 0, None, open=True, text='标签类别',values=('root')) # 树视图添加根节点
tree.bind('<Button-1>', self.lable_tree_click) # 绑定单击离开事件
tree.bind('<Button-3>',self.lable_tree_Rclick)
# 插入新的标签到标签树
def insert_label_tree(self,tree,lable_list):
tree.insert(self.tr_root, 0, None, text=lable_list, values=(lable_list)) # 添加二级节点
def lable_tree_click(self,event): # 单击
for item in self.lable_tree.selection():
item_text = self.lable_tree.item(item, "values")
if item_text[0] == 'root':
continue
print(item_text[0])
#lable 树右键点击事件
def lable_tree_Rclick(self,event):
for item in self.lable_tree.selection():
item_text = self.lable_tree.item(item, "values")
if item_text[0] == 'root':
print('已经点击了右键' + 'root')
else:
print('已经点击了右键')
def initButton(self):
lable = tk.Label(self.frm_control, font='Times 9 bold', text="控制界面", fg='#4876FF', anchor='nw')
lable.grid(row=0, column=0, sticky=tk.NSEW,padx=6,pady=6)
self.set_work_dir = ttk.Button(self.frm_control,text='设置目录',command=self.set_work_dir)
self.set_work_dir.grid(row=1, column=0, padx=5,sticky=tk.NSEW)
self.next_image = ttk.Button(self.frm_control,text='下一张',state=tk.DISABLED,command=self.get_next_image)
self.next_image.grid(row=1, column=1, padx=5,sticky=tk.NSEW)
self.pre_image = ttk.Button(self.frm_control, text='上一张',state=tk.DISABLED,command=self.get_pre_image)
self.pre_image.grid(row=1, column=2, padx=5,sticky=tk.NSEW)
self.revoke = ttk.Button(self.frm_control, text='撤销标注',state=tk.DISABLED,command=self.remove_canvas_lable_all)
self.revoke.grid(row=1, column=3, padx=5,sticky=tk.NSEW)
self.label_rect = ttk.Button(self.frm_control, text='创建矩形',state=tk.DISABLED,command=self.create_label_rect)
self.label_rect.grid(row=2, column=0, padx=5,sticky=tk.NSEW)
self.label_scat = ttk.Button(self.frm_control, text='创建散点',state=tk.DISABLED,command=self.create_label_scat)
self.label_scat.grid(row=2, column=1, padx=5,sticky=tk.NSEW)
self.save_lable = ttk.Button(self.frm_control, text='保存标注', state=tk.DISABLED,command=self.save_lable_file)
self.save_lable.grid(row=2, column=2, padx=5,sticky=tk.NSEW)
# 删除全部画布上的标注图形
def remove_canvas_lable_all(self):
self.canvas.delete(tk.ALL)
if len(self.image_paths) < 1:
return
self.showImage(self.image_paths[self.image_index])
self.scatter_point.clear()
self.number = 0
# self.image_index = self.image_index + 1
#self.next_image.config(state=tk.ACTIVE)
# Radiobutton选择时回调
def radiobutton_change(self):
#print(self.lable_Type.get())
self.number = 0
#设置工作目录
def set_work_dir(self):
work_dir = dir.Directory()
temp_dir = work_dir.show(initialdir='.', title='设置工作目录')
if temp_dir != '':
self.work_dir = temp_dir
self.get_all_img(self.work_dir)
if len(self.image_paths) < 1:
mb.showinfo("提醒","当前目录没有图片！")
return
self.showImage(self.image_paths[self.image_index])
#self.image_index = self.image_index + 1
self.next_image.config(state=tk.ACTIVE)
self.create_image_tree(self.image_name_tree, self.work_dir)
self.revoke.config(state=tk.ACTIVE)
self.label_rect.config(state=tk.ACTIVE)
self.label_scat.config(state=tk.ACTIVE)
self.save_lable.config(state=tk.ACTIVE)
else:
mb.showinfo("提醒", "目录选择失败！")
#下一张图片
def get_next_image(self):
if len(self.image_paths) < 1:
mb.showerror("错误", "当前目录中没有图片，请重新设置目录！")
return
if len(self.image_paths) - 1 < self.image_index + 1:
mb.showinfo('提醒', '已经到最后一张！')
self.next_image.config(state=tk.DISABLED)
return
if self.image_index == 0:
self.pre_image.config(state=tk.ACTIVE)
self.canvas.delete(tk.ALL)
names = self.image_paths[self.image_index + 1]
self.showImage(names)
self.image_index = self.image_index + 1
self.x = 0
self.y = 0
#前一张图片
def get_pre_image(self):
if self.image_index < 1:
mb.showinfo("提醒", "已经是第一张照片了！")
self.pre_image.config(state=tk.DISABLED)
return
if self.image_index == len(self.image_paths) - 1:
self.next_image.config(state=tk.ACTIVE)
names = self.image_paths[self.image_index - 1]
self.showImage(names)
self.image_index = self.image_index - 1
self.x = 0
self.y = 0
#获取工作目录下面的所有图像名称
def get_all_img(self,path):
file_list = os.listdir(path)
for i in range(len(file_list)):
name = path + '/' + file_list[i]
kind = filetype.guess(name)
if kind and kind.extension in IMAGE_FILE_TYPE:
self.image_paths.append(name)
self.image_names.append(file_list[i])
#在canvas上显示图片
def showImage(self,image_path):
global img
img = Image.open(image_path)
w, h = img.size
img = self.resize(img,w,h,800,600)
w, h = img.size
img = ImageTk.PhotoImage(img)
self.canvas.configure(width=w, height=h)
self.canvas.create_image(0,0,image=img,anchor='nw', tags='show_image')
# 监听鼠标在canvas上的移动事件
def mouse_on_canvas(self,event):
if self.lable_Type == 1 and self.work_dir != '' and self.clicked:
self.canvas.delete('mouse_move_paint_line' + str(self.number - 1))
self.canvas.delete('mouse_move_paint_rect' + str(self.number - 1))
self.canvas.create_line((self.x,self.y),(event.x,event.y),width=3,fill='red', tags='mouse_move_paint_line' + str(self.number))
self.canvas.create_rectangle((self.x,self.y),(event.x,event.y),width=3, fill='#000', outline='green', stipple='gray12',tags='mouse_move_paint_rect' + str(self.number))
self.number = self.number + 1
elif self.lable_Type == 2 and self.work_dir != '' and len(self.scatter_point) > 0 and self.clicked:
last_p = self.scatter_point[-1]
self.canvas.delete('2_mouse_move_paint_line' + str(self.number - 1))
self.canvas.create_line((last_p[0], last_p[1]), (event.x, event.y), width=3,tags='2_mouse_move_paint_line' + str(self.number))
self.number += 1
if len(self.scatter_point) > 1 and self.scatter_point[0][0] < event.x + 5 and self.scatter_point[0][
0] > event.x - 5 and self.scatter_point[0][1] < event.y + 5 and self.scatter_point[0][1] > event.y - 5:
first_p = self.scatter_point[0]
self.canvas.create_oval((first_p[0] - 10, first_p[1] - 10, first_p[0] + 10, first_p[1] + 10), fill='green', tags='oval_exp')
else:
self.canvas.delete('oval_exp')
#点击鼠标左键监听事件，用于获取点击鼠标的位置（x,y）,没有使用
def left_mouse_click(self,event):
self.canvas.create_oval((event.x - 5, event.y - 5, event.x + 5, event.y + 5), fill='red',tags='oval')
if self.lable_Type == 1:
if self.clicked:
self.clicked = False
r = simpledialog.askstring('标签', '请输入标签', initialvalue='浙AXXXX')
if self.image_index >= 0 and r != None:
if r not in self.image_label:
self.insert_label_tree(self.lable_tree,r)#添加标签数
self.image_label.append(r)
self.lable_save_file.create_object(self.image_paths[self.image_index], 'image', 'path')
self.lable_save_file.set_lable_info(r, lable_name='palte')
self.lable_save_file.set_position_rect('poiston', w=self.factor * (event.x - self.x),
h=self.factor * (event.y - self.y), x=self.factor * self.x,
y=self.factor * self.y)
else:
self.x = event.x;
self.y = event.y
self.clicked = True
elif self.lable_Type == 2:
if not self.clicked:
self.scatter_point.clear()
self.clicked = True
self.scatter_point.append([event.x, event.y])
else:
last_p = self.scatter_point[-1]
self.canvas.create_line((last_p[0], last_p[1]), (event.x, event.y), fill='green', width=3, tags='line')
if len(self.scatter_point) > 0 and self.scatter_point[0][0] < event.x + 5 and self.scatter_point[0][
0] > event.x - 5 and self.scatter_point[0][1] < event.y + 5 and self.scatter_point[0][1] > event.y - 5:
self.clicked = False
r = simpledialog.askstring('标签', '请输入标签', initialvalue='浙AXXXX')
if self.image_index >= 0 and r != None:
info = {'lt': self.scatter_point[0], 'rt': self.scatter_point[1], 'lb': self.scatter_point[2],
'rb': self.scatter_point[3]}
self.lable_save_file.create_object(self.image_paths[self.image_index], 'image', 'path')
self.lable_save_file.set_lable_info(r, lable_name='palte')
self.lable_save_file.set_position_scat('poiston', info)
self.scatter_point.clear()
else:
self.scatter_point.append([event.x, event.y])
# 创建矩形label按钮的回调方法
def create_label_rect(self):
self.lable_Type = 1
self.canvas.delete(tk.ALL)
if len(self.image_paths) < 1:
return
self.showImage(self.image_paths[self.image_index])
self.label_scat.config(state=tk.DISABLED)
# 创建散点label按钮的回调方法
def create_label_scat(self):
self.lable_Type = 2
self.canvas.delete(tk.ALL)
if len(self.image_paths) < 1:
return
self.showImage(self.image_paths[self.image_index])
self.label_rect.config(state=tk.DISABLED)
#保存标签
def save_lable_file(self):
self.lable_save_file.save()
self.label_rect.config(state=tk.ACTIVE)
self.label_scat.config(state=tk.ACTIVE)
#重置大小
def resize(self,pil_image,w, h, w_box=0, h_box=0):
if w_box == 0 or h_box == 0:
w_box = w
h_box = h
f1 = 1.0 * w_box / w
f2 = 1.0 * h_box / h
self.factor = min([f1, f2])
width = int(w * self.factor)
height = int(h * self.factor)
return pil_image.resize((width, height), Image.ANTIALIAS)
# 设置主界面相关属性
def set_mainUI(root):
root.title('图片标注工具')
root.option_add("*Font", "宋体")
center_window(root, 960, 640)
#root.maxsize(960, 640)
root.minsize(640,480)
root.iconbitmap('app.ico')
root.config(background='SteelBlue')
app = MainGUI(root)
root.mainloop()
#获取屏幕大小
def get_screen_size(window):
return window.winfo_screenwidth(), window.winfo_screenheight()
#获取串口大小
def get_window_size(window):
return window.winfo_reqwidth(), window.winfo_reqheight()
#设置串口居中
def center_window(root, width, height):
screenwidth = root.winfo_screenwidth()
screenheight = root.winfo_screenheight()
size = '%dx%d+%d+%d' % (width, height, (screenwidth - width) / 2, (screenheight - height) / 2)
root.geometry(size)
if (__name__ == '__main__'):
root = tk.Tk()
set_mainUI(root)这个项目已经上传到github，如果有需要的朋友们，可以从github clone下来。
地址：https://github.com/YongGangZ/LabelTool.git
希望大家多多提出一些建议！
版权声明：本文为zhangyonggang886原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。