【发布时间】:2020-02-10 03:58:24
【问题描述】:
请我需要解决方案来更正和运行 mask rcnn 的此代码
【问题讨论】:
-
不要发布代码图片,而是发布代码。
标签: machine-learning deep-learning computer-vision faster-rcnn
请我需要解决方案来更正和运行 mask rcnn 的此代码
【问题讨论】:
标签: machine-learning deep-learning computer-vision faster-rcnn
正如 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"])
更多信息请参考this issue。
【讨论】: