machine-learning - 如何通过测试我的掩码 rcnn 模型来纠正此错误

Question

请我需要解决方案来纠正和运行掩码 rcnn 的此代码

标签： machine-learningdeep-learningcomputer-visionfaster-rcnn

Answer 1

正如 Gtomika 所评论的，请以文本格式提供错误跟踪。还提供有关您从中获取模型的存储库的更多详细信息以及您认为相关的任何其他信息。

根据我过去的经验，我很确定您使用的是 Matterplot 的掩码 RCNN，而您的问题是由于类数不匹配造成的。您正在尝试加载在不同类数上训练的模型的权重。您应该排除一些层，例如“mrcnn_bbox_fc”、“mrcnn_class_logits”。

修复是更改model.load_weights(COCO_MODEL_PATH, by_name=True)为model.load_weights(filepath, by_name=True, exclude=[ "mrcnn_class_logits", "mrcnn_bbox_fc", "mrcnn_bbox", "mrcnn_mask"])

请参阅此问题以获取更多信息。