推送者seo/网站关键词推广优化

Keras文档称它返回“N Numpy预测数组”.

在496个带有4个类的图像示例中使用它,我得到一个4维数组

(496,4,4,512).其他2个维度是什么？

最后,我希望有一个X(示例)数组和一个Y(标签)数组.

img_width, img_height = 150, 150

top_model_weights_path = 'bottleneck_fc_model.h5'

train_data_dir = 'data/train'

validation_data_dir = 'data/validation'

nb_train_samples = 496

nb_validation_samples = 213

epochs = 50

batch_size = 16

number_of_classes = 3

datagen = ImageDataGenerator(rescale=1. / 255)

# build the VGG16 network (exclude last layer)

model = applications.VGG16(include_top=False, weights='imagenet')

# generate training data from image files

train_generator = datagen.flow_from_directory(

train_data_dir,

target_size=(img_width, img_height),

batch_size=batch_size,

class_mode='categorical',

shuffle=False)

# predict bottleneck features on training data

bottleneck_features_train = model.predict_generator(

train_generator, nb_train_samples // batch_size)

print(bottleneck_features_train.shape)

train_data = np.load(open('bottleneck_features_train.npy', 'rb'))

print(train_data.shape)