MaskRcnn模型不会从“ last”开始训练

时间:2019-06-07 15:42:52

标签: keras conv-neural-network

我正在尝试在黑色素瘤图像上训练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

0 个答案:

没有答案