一、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 = []