python如何将灰度图片转为数组并储存到csv文件中

首先,我们知道要是图片的大小为80*60,生成的数据就会有80*60个,这样要是图片多的话,csv得到数据就是很多,我们可以考虑把图片缩小

我的图片都是放在目录E:\machineStudy\ta-lib\picture下

python如何将灰度图片转为数组并储存到csv文件中

python如何将灰度图片转为数组并储存到csv文件中 

我们将灰度图片转为csv,一般都是用来做神经网络学习,一般我们都是给不同的情况贴上标签,我这里就是将10中不同的情况贴上0-9的标签,0-9目录下存在不同情况的图片,我们要做的就是将每个目录下的灰度图都转为csv格式数据,一个图片对应一行数据,并且在csv文件中的第一位存放着我们的标签,也就是0-9的数字。这里要注意,我一开始就说了,我们灰度图的图片可能会很大,我的就是80*60,csv得到的数据会很多,打开文件都得花很长时间,所以我们就可以考虑将我们的图片放小。

程序代码:

# -*- coding:utf-8 -*-
import csv
import os
import cv2
#图片所在的目录
IMG_DIR = r"E:\machineStudy\ta-lib\picture"


def convert_img_to_csv(img_dir):
    #这个是输出文件的位置,自己定义
    with open(r'E:\machineStudy\ta-lib\picture\train_data.csv', 'w',
              newline='') as f:
        column_name = ['label']
        #这里图片的大小为80*60,生成的数据就会有80*60个,csv文件一行是放不下的,所以我们要把图片放小
        column_name.extend('pixel%d' % i for i in range(80 * 60))

        writer = csv.writer(f)
        writer.writerow(column_name)

        for i in range(10):
            img_file_path = os.path.join(img_dir, str(i))
            img_list = os.listdir(img_file_path)
            print(img_file_path)

            for img_name in img_list:
                img_path = os.path.join(img_file_path, img_name)
                img = cv2.imread(img_path, cv2.IMREAD_GRAYSCALE)
                #下面两行是将图片放小为原来的1/8,
                x, y = img.shape[0:2]
                img_test1 = cv2.resize(img, (int(y / 8), int(x / 8)))

                image_data = [i]
                image_data.extend(img_test1.flatten())
                # print(image_data)
                writer.writerow(image_data)


if __name__ == "__main__":
    convert_img_to_csv(IMG_DIR)

控制台输出遍历完的文件目录名

python如何将灰度图片转为数组并储存到csv文件中

 

查看生成的train_data.csv文件

python如何将灰度图片转为数组并储存到csv文件中

第一列就是我们不同的标签,其他列就是我们图片的像素,一行就是一个图片的数据

 

      

上一篇:代理服务器与负载均衡


下一篇:Linux代理与负载均衡