读取图片信息并存储数据为CSV文件

下面介绍一种读取图片,并将图片信息保存为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’]) 还可以加标签

你可能感兴趣的:(图像处理)