我正在尝试使用经过预先训练的模型,专门训练徽标。我正在使用MobileNet进行徽标培训。如果我们从头开始制作CNN模型,那么我们会在其中提供标签。我不知道如何在转学中提供标签。当我们使用flow_from_directory函数时,图像数据生成器都会自动提供标签。很少的代码如下所示。细说!
training_set = train_datagen.flow_from_directory('Datasets/Train',
target_size = (224, 224),
batch_size = 32,
class_mode = 'categorical')
r = model.fit_generator(training_set,validation_data=test_set,epochs=5,steps_per_epoch=len(training_set),
validation_steps=len(test_set)
)
我相信标签是从目录架构中推断出来的,所以如果您的主目录对于每个训练和测试集都是这样的:
main_directory/
...class_a/
......a_image_1.jpg
......a_image_2.jpg
...class_b/
......b_image_1.jpg
......b_image_2.jpg
你应该没事的。有关更多示例,请参考此
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句