图片数据转换为csv文件(附python代码)

深度学习 同时被 2 个专栏收录
26 篇文章 2 订阅
18 篇文章 0 订阅

首先抛出两个问题:

· 什么是csv文件?

· 为什么要用csv文件?

CSV文件:CSV逗号分隔值文件(Comma Separated value),是一种用来存储数据的纯文本文件格式,通常用于电子表格或数据库软件。
CSV文件的好处:
1.文件结构简单,基本上和文本的差别不大;
2.可以和Excle进行转换,这是一个很大的优点,很容易进行察看模式转换,但是如果你同样的csv文件和将其转换成xls文件后的size比较就更加明白它在size上的优势了。
3.由于其简单的存储方式,一方面可以减少存储信息的容量,这样有利于网络传输以及客户端的再处理;同时由于是一堆没有任何说明的数据,具备基本的安全性。

在深度学习的使用中,一定会遇到将自己的图片数据集转化为csv文件,下面直接上干货。

#coding:utf-8
import os
import csv

def create_csv(dirname):
    path = './dataset/'+ 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(['./dataset/'+str(dirname) +'/'+ str(n),'./dataset/' + str(dirname) + 'label/' + str(n[:-4] + '.png')])
            else:
                pass

if __name__ == "__main__":
    create_csv('train')
    create_csv('test')
  • 8
    点赞
  • 14
    评论
  • 60
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值