Python CSV标签生成脚本
CSV 文件生成的 Python 源码，适合要批量图片标签的朋友。用 Python 的csv模块，几行代码就能把数据写成标准的 CSV 格式，拿去训练模型也省心。
Python 的 csv 模块属于标准库，写起来不费劲。常用的csv.writer就挺顺手，搭配with open()写法，文件读写也不用你操心太多。
拿京东猪脸识别比赛举例，通常一张图配一个标签，比如image1.jpg对应猪 1 类，直接用writer.writerow()一行一行写进去就完事儿了。
代码结构也清晰，基本套路是：
import csv
with open('output.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerow(['文件名', '标签'])
writer.writerow(['image1.jpg', '猪 1 类'])
writer.writerow(['image2.jpg', '猪 2 类'])
有时候你还会遇到带 header 的需求，那就在第一行写上字段名就好。如果数据多，可以配合 pandas 一起搞，效率高不少。
如果你也在做图像分类或者数据导出这类事儿，这段源码蛮值得收藏。想了解更多，你可以看看这几篇相关文章：
哦对了，写 CSV 的时候别忘了newline=''，不然跨平台时多出空行，这坑我之前也踩过。
文件大小：1.15KB
评论区