Machine learning 我们如何为迁移学习培训设置标签？
我正在尝试使用一个预先培训过的模型，专门针对logo进行培训。我正在使用MobileNet进行标识培训。如果我们从头开始制作CNN模型，那么我们会在模型中提供标签。我不知道如何在迁移学习中提供标签。当我们使用flow_from_directory函数时，任一图像数据生成器都会自动提供标签。代码的一小部分如下所示。精心设计Machine learning 我们如何为迁移学习培训设置标签？,machine-learning,deep-learning,neural-network,conv-neural-network,transfer-learning,Machine Learning,Deep Learning,Neural Network,Conv Neural Network,Transfer Learning,我正在尝试使用一个预先培训过的模型，专门针对logo进行培训。我正在使用MobileNet进行标识培训。如果我们从头开始制作CNN模型，那么我们会在模型中提供标签。我不知道如何在迁移学习中提供标签。当我们使用flow_from_directory函数时，任一图像数据生成器都会自动提供标签。代码的一小部分如下所示。精心设计 training_set = train_datagen.flow_from_directory('Datasets/Train',
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