如果没有data数据,只有本地文件夹下面的图片,这段代码应该怎么改:data = np.load('data.npy') x = data[:, :-1] y = data[:, -1]
时间: 2023-06-06 12:07:37 浏览: 209
Matlab-在matlab中读+写Numpy.NPY文件实现.zip
如果没有data数据,只有本地文件夹下面的图片,你需要用opencv库的imread()函数读取这些图片并将其转换为numpy数组。你可以按照以下示例代码来修改:
import os
import cv2
import numpy as np
data = []
labels = []
path = 'your_folder_path'
for folder in os.listdir(path):
for file in os.listdir(os.path.join(path, folder)):
img = cv2.imread(os.path.join(path, folder, file))
img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) #如果是彩色图像需要将其转化为灰度图
img = cv2.resize(img,(img_size,img_size)) #如果需要调整图像大小,可以用cv2.resize函数
data.append(np.array(img))
labels.append(folder)
data = np.array(data) / 255.0 #归一化
labels = np.array(labels)
x = data
y = labels
注意:你需要根据自己的需求修改上述代码。
阅读全文