#卷积神经网络#安装Theano
#安装Tensorflow
#安装Keras
为了使用 flow_from_directory 。您必须具有以下文件夹结构。
flow_from_directory
./Dataset/ ./Train/ ../Folder_1/ ../img_1.jpg ../img_2.jpg ............ ../Folder_2/ ../img_1.jpg ../img_2.jpg
其中Folder_i包含类i的图像。
在你的道路上 E:\Major Project\Data 你必须有 n 每个对应于每个类的文件夹。
E:\Major Project\Data
n
然后你可以打电话 flow_from_directory 如
train_datagen.flow_from_directory('E:\Major Project\Data\',target_size = (64, 64),batch_size = 32,class_mode = 'categorical')
你会得到这样的输出
Found xxxx images belonging to yyyy classes
如果其他一切都正确,模型将开始训练
训练后如果你想使用 flow_from_directory 在...的帮助下做出预测 predict_generator 你可以这样做
predict_generator
您可以将flow_from_directory中的batch_size值从默认值(即batch_size = 32)更改为batch_size = 1。然后将predict_generator的步骤设置为测试图像的总数。
test_datagen = ImageDataGenerator(rescale=1./255) test_generator = test_datagen.flow_from_directory( test_dir, target_size=(200, 200), color_mode="rgb", shuffle = False, class_mode='categorical', batch_size=1) filenames = test_generator.filenames nb_samples = len(filenames) predict = model.predict_generator(test_generator,steps = nb_samples)
如果要在单个图像上进行预测。
import cv2 import numpy as np img = cv2.imread('path_to_file') img = cv2.resize(img, (64, 64)) img = img.reshape(1, 64, 64, 3) model.predict(img)