我正在尝试在黑色素瘤图像上训练maskRcnn模型。我最初只训练网络负责人,然后才是整个网络。 但是,该代码仅训练网络头。
model.train(dataset_train,dataset_val,
learning_rate=config.LEARNING_RATE,
epochs=30,
layers='heads')
后续代码:
model.train(dataset_train,dataset_val,
learning_rate=config.LEARNING_RATE,
epochs=30,
layers='all')
不训练模型,也不给出错误。执行就完成了。可以找到整个脚本here