下面介绍一种读取图片,并将图片信息保存为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"C:/pypractise/cnn/cnn/test")
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,128))
save = pd.DataFrame(data)
save.to_csv('c:/pypractise/cnn/cnn/%d.csv' %i, index=False, header=False)
存储的路径以及名字
#index=False,header=False表示不保存行索引和列标题
save = pd.DataFrame(np_data, columns = [‘year’, ‘month’, ‘day’]) 还可以加标签