如何通过测试我的 mask rcnn 模型来更正此错误
How can I correct this error from testing of my mask rcnn model
我需要更正的解决方案和运行掩码rcnn的代码
正如 Gtomika 所说，请提供文本格式的错误跟踪。还请提供有关从中获取模型的存储库的更多详细信息以及您认为相关的任何其他信息。
根据我过去的经验，我很确定您使用的是 Matterplot 的 mask RCNN，您的问题是由于 class 计数不匹配造成的。您正在尝试加载在不同 class 计数下训练的模型的权重。您应该排除一些图层，例如 '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"])
参考this issue了解更多信息。
我需要更正的解决方案和运行掩码rcnn的代码
正如 Gtomika 所说，请提供文本格式的错误跟踪。还请提供有关从中获取模型的存储库的更多详细信息以及您认为相关的任何其他信息。
根据我过去的经验，我很确定您使用的是 Matterplot 的 mask RCNN，您的问题是由于 class 计数不匹配造成的。您正在尝试加载在不同 class 计数下训练的模型的权重。您应该排除一些图层，例如 '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"])
参考this issue了解更多信息。