做深度学习的时候，想把labelme的json文件改成yolo的txt文件。
https://github.com/ivder/LabelMeYoloConverter/blob/master/convert.py
我把它转过来指着这边。
Traceback (most recent call last):
File "convert.py", line 67, in <module>
x1 = float(lines[idx+5].rstrip(','))
IndexError: list index out of range
有这样的错误。
我问这个问题是因为无论我怎么看，我都无法分辨出问题的哪一部分。
下面是代码 convert.py。
# -*- coding: utf-8 -*-
'''
LabelMe JSON format -> YOLO txt format
save dataset (학습 자료) in dataset/
output will be saved in result/
JSON format will be moved to json_backup/
Finally, please manually copy text file together with image into 1 folder. (Easier to maintain)
마지막으로 txt파일이랑 이미지파일이랑 같은 폴더에 복사하세요 (관리하기 위한 쉬움)
'''
import os
from os import walk, getcwd
from PIL import Image
def convert(size, box):
dw = 1./size[0]
dh = 1./size[1]
x = (box[0] + box[1])/2.0
y = (box[2] + box[3])/2.0
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)
"""-------------------------------------------------------------------"""
""" Configure Paths"""
mypath = "./dataset/"
outpath = "./result/"
json_backup ="./json_backup/"
wd = getcwd()
#list_file = open('%s_list.txt'%(wd), 'w')
""" Get input json file list """
json_name_list = []
for file in os.listdir(mypath):
if file.endswith(".json"):
json_name_list.append(file)
""" Process """
for json_name in json_name_list:
txt_name = json_name.rstrip(".json") + ".txt"
""" Open input text files """
txt_path = mypath + json_name
print("Input:" + txt_path)
txt_file = open(txt_path, "r", encoding="UTF-16")
""" Open output text files """
txt_outpath = outpath + txt_name
print("Output:" + txt_outpath)
txt_outfile = open(txt_outpath, "a", encoding="UTF-16")
""" Convert the data to YOLO format """
lines = txt_file.read().split('\r\n') #for ubuntu, use "\r\n" instead of "\n"
for idx, line in enumerate(lines):
if ("lineColor" in line):
break #skip reading after find lineColor
if ("label" in line):
x1 = float(lines[idx+5].rstrip(','))
y1 = float(lines[idx+6])
x2 = float(lines[idx+9].rstrip(','))
y2 = float(lines[idx+10])
cls = line[16:17]
#in case when labelling, points are not in the right order
xmin = min(x1,x2)
xmax = max(x1,x2)
ymin = min(y1,y2)
ymax = max(y1,y2)
img_path = str('%s/dataset/%s.jpg'%(wd, os.path.splitext(json_name)[0]))
im=Image.open(img_path)
w= int(im.size[0])
h= int(im.size[1])
print(w, h)
print(xmin, xmax, ymin, ymax)
b = (xmin, xmax, ymin, ymax)
bb = convert((w,h), b)
print(bb)
txt_outfile.write(cls + " " + " ".join([str(a) for a in bb]) + '\n')
os.rename(txt_path,json_backup+json_name) #move json file to backup folder
enter code here