赞
踩
地址:https://github.com/tzutalin/labelImg
很多教程里面提到的是需要使用Anocanda进行安装,题主这里使用比较简单的python第三方库下载安装工具 pip
注意,在下载的过程中会出现速度过慢的情况,建议更换下载源(题主在使用原本的默认源的时候出现了read time out的问题),如何更换下载源可以参考题主的另一篇博客
当然不想看的,可以直接采用题主下面的安装代码。
pip install lxml -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install pyqt5 -i https://pypi.tuna.tsinghua.edu.cn/simple/
将我们刚刚的labelImg-master文件夹里面的resources.py文件移动到libs里面,不然会出现找不到模板的bug
cmd移动到labelImg-master文件夹所在的文件夹,运行labelImg.py 文件(因为题主电脑有py2和py3,所以把python3的启动变成了python3),如果电脑只有一个python版本的话直接运行python labelImg.py就行了。
在使用的过程中,不想要XML类型的文件,例如想要txt类型的文件。也是可以通过python代码来实现的,这里贴一个XML转换成为txt文件的代码
import xml.etree.ElementTree as ET import pickle import os from os import listdir, getcwd from os.path import join sets=['train', 'val'] #替换为自己的数据集 classes = ["person", "hat"] #修改为自己的类别 def convert(size, box): dw = 1./(size[0]) dh = 1./(size[1]) x = (box[0] + box[1])/2.0 - 1 y = (box[2] + box[3])/2.0 - 1 w = box[1] - box[0] h = box[3] - box[2] x = x*dw w = w*dw y = y*dh h = h*dh return (x,y,w,h) def convert_annotation(image_id): try: in_file = open('label/%06d.xml'%(image_id)) except: return 0 out_file = open('train2017/%06d.txt'%(image_id), 'w') tree=ET.parse(in_file) root = tree.getroot() size = root.find('size') w = int(size.find('width').text) h = int(size.find('height').text) for obj in root.iter('object'): difficult = obj.find('difficult').text cls = obj.find('name').text #如果不是人或者是难以识别的人,跳过 if cls not in classes or int(difficult)==1: continue cls_id = classes.index(cls) xmlbox = obj.find('bndbox') b = (float(xmlbox.find('xmin').text), float(xmlbox.find('xmax').text), float(xmlbox.find('ymin').text), float(xmlbox.find('ymax').text)) bb = convert((w,h), b) out_file.write(str(cls_id) + " " + " ".join([str(a) for a in bb]) + '\n') return 1 wd = getcwd() if not os.path.exists('labels'): os.makedirs('labels') list_file = open('train.txt', 'w') for image_id in range(148,149): if (convert_annotation(image_id) == 1): list_file.write('../coco/images/train2017/%06d.jpg\n'%(image_id)) #print('train/%06d.jpg\n'%(i)) list_file.close() print('运行结束')
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。