python读取图片转cvs_python将图片转换为csv

本篇文章主要介绍如何将图片转为csv文件进行保存，主要利用python内置的库csv。

import csv,os,cv2

def convert_img_to_csv(img_dir):

#设置需要保存的csv路径

with open(r"D:\dataset\kaggle\MNIST\enhance.csv","w",newline="")

as f:

#设置csv文件的列名

column_name = ["emotion"]

column_name.extend("pixel")

#将列名写入到csv文件中

writer = csv.writer(f)

writer.writerow(column_name)

#该目录下有9个目录,目录名从0-9

for i in range(10):

#获取目录的路径

img_temp_dir = os.path.join(img_dir,str(i))

#获取该目录下所有的文件

img_list = os.listdir(img_temp_dir)

#遍历所有的文件名称

for img_name in img_list:

#判断文件是否为目录,如果为目录则不处理

if not os.path.isdir(img_name):

#获取图片的路径

img_path = os.path.join(img_temp_dir,img_name)

#因为图片是黑白的，所以以灰色读取图片

img = cv2.imread(img_path,cv2.IMREAD_GRAYSCALE)

#图片标签

row_data = [i]

#获取图片的像素

row_data.extend(img.flatten())

#将图片数据写入到csv文件中

writer.writerow(row_data)

if __name__ == "__main__":

#将该目录下的图片保存为csv文件

convert_img_to_csv(r"D:\python_workspace\kaggle\DigitRecongnizer\LeN

et\img_enhance")1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

修改

for img_name in img_list:

#图片路径

img_path = os.path.join(img_temp_dir,img_name)

#判断文件是否为目录,如果为目录则不处理

if not os.path.isdir(img_path):

#因为图片是黑白的，所以以灰色读取图片

img = cv2.imread(img_path,cv2.IMREAD_GRAYSCALE)

#图片标签

row_data = [i]

#获取图片的像素

row_data.extend(img.flatten())

#将图片数据写入到csv文件中

writer.writerow(row_data)1

2

3

4

5

6

7

8

9

10

11

12

13