import re
file_path = './tile_data_1011n1/train.txt'
new_file_path_base = './tile_data_1011n1/label/'
with open(file_path, 'r') as file_to_read:
while True:
line = file_to_read.readline()
if not line:
break
else:
print(line)
line = list(filter(None, line.split(" ")))
box_num = int((len(line)-4)/5) # 每一行的 gt_box 个数,对应的就是新的 txt 文件的行数
pic_name = re.split(" |/|\.", line[1])
# 打开新的 txt 文件,准备写入
new_file_name = pic_name[5]
new_file = open(new_file_path_base + new_file_name + '.txt', 'w')
one_line = ""
for i in range(box_num):
idx = 4 + i * 5
one_line += line[idx]
for j in range(1, 5):
one_line += ' '
one_line += line[idx + j]
one_line += '\n'
new_file.write(one_line)
new_file.close()
yolov3_to_yolov5
最新推荐文章于 2023-12-04 15:39:18 发布
956

被折叠的 条评论
为什么被折叠?



