图片数据转换为csv

qwe7150849

于 2021-07-13 15:34:53 发布

阅读量2.9k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/qwe7150849/article/details/118701618

该代码段用于从指定目录下读取.jpg图片文件，并生成对应的CSV文件，其中包含图片路径和对应的标签路径。它遍历指定的'images'子目录，将图片名及其对应的标签.png文件名写入CSV文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os
import csv
 
def create_csv(dirname):
    path='./chongqing1_round1_testA_20191223/'+dirname+'/'
    name=os.listdir(path)
    with open (dirname+'.csv','w') as csvfile:
        writer=csv.writer(csvfile)
        writer.writerow(['data','label'])
        for n in name:
            if n[-4:]=='.jpg':
                print(n)
                writer.writerow(['./chongqing1_round1_testA_20191223/'+str(dirname)+'/'+str(n),'./chongqing1_round1_testA_20191223/' + str(dirname) + 'label/' + str(n[:-4] + '.png')])
            else:
                pass
if __name__ == "__main__":
    create_csv('images')