python读取图片并储存为CSV文件

读取图片的标签并储存到CSV文件，代码如下：

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

输出结果：

储存图像信息并保存到CSV文件。

import cv2
import os
import numpy as np
import pandas as pd

def get_imlist(path):
    return [os.path.join(path,f) for f in os.listdir(path) if f.endswith('.jpg')]

c=get_imlist(r'./chongqing1_round1_testA_20191223/images')
d=len(c)    # 图像个数
# 遍历每张图片
for i in range(d):
    img = cv2.imread(c[i],cv2.IMREAD_GRAYSCALE)  # 打开图像
    img_ndarray = np.asarray(img, dtype='float64') / 256  # 将图像转化为数组并将像素转化到0-1之间
    data = cv2.resize(img_ndarray,(128,1))
    save = pd.DataFrame(data)
    save.to_csv('images.csv', index=False, header=False)