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

ldxq2e6h  于 2021-09-29  发布在  Java
关注(0)|答案(0)|浏览(556)

输入数据CSV文件
每一行都应该转换成txt文件,一些图像超过3或2个边界框,因此表示该文件的txt文件应该包含大量的行值
我编写了一些代码,但它只在csv文件的行中为每个txt文件提供一行值

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()

根据每个图像边界框,我们需要相应的txt文件

暂无答案!

目前还没有任何答案,快来回答吧!

相关问题