python将灰度图像像素存到.csv文件

最新推荐文章于 2023-02-15 13:38:54 发布

despacito,

最新推荐文章于 2023-02-15 13:38:54 发布

阅读量2.9k

点赞数

分类专栏：图像处理

本文链接：https://blog.csdn.net/qq_40108803/article/details/103203156

版权

本文介绍了如何使用Python将灰度图像的像素数据存储到CSV文件中，包括训练集（train file）和测试集（test file）的生成过程。

摘要由CSDN通过智能技术生成

一、python代码：

import csv
import os
import cv2


IMG_DIR = r"C:\Users\tianr\Desktop\Experiment _Data\GrayImg150(64x64)"


def convert_img_to_csv(img_dir):

    with open(r'C:\Users\tianr\PycharmProjects\DeepLearn\Classical_CNNS\LeNet-5\BreastDatas_test.csv', 'w', newline='') as f:
        # column_name = ['label']
        column_name = []
        column_name.extend('pixel%d' % i for i in range(64*64))

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

        for i in range(2):
            img_file_path = os.path.join(img_dir, str(i))
            img_list = os.listdir(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)
                # image_data = [i]
                image_data = []