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 发布