将 .csv 文件转换为 yolo 格式 (txt)

时间:2021-07-27 12:51:06

标签: python object-detection yolov5

输入数据 csv file

每一行都应该转换成 txt 文件,一些图像超过 3 或 2 个边界框,所以代表该文件的 txt 文件应该包含那个糊状行值

我写了一些代码,但它只在每个 txt 文件中从 csv 文件的行中给出一个行值

for index, row in data.iterrows():
    file_name ='{0}.txt'.format(row['name']) 
    with open(file_name, 'w') as f:
        f.write(str(row['class']))
        f.write(' ')
        f.write(str(row['x-axis']))
        f.write(' ')
        f.write(str(row['y-axis']))
        f.write(' ')
        f.write(str(row['width']))
        f.write(' ')
        f.write(str(row['height']))
        f.close()    

根据每个图片的bounding box我们需要对应的txt文件

0 个答案:

没有答案