如何通過測試我的面罩 rcnn model 來糾正此錯誤

Question

請我需要解決方案來糾正和運行掩碼 rcnn 的此代碼

Answer 1

正如 Gtomika 所評論的，請以文本格式提供錯誤跟蹤。 還提供有關您從中獲得 model 的存儲庫的更多詳細信息以及您認為相關的任何其他信息。

根據我過去的經驗，我很確定您使用的是 Matterplot 的掩碼 RCNN，而您的問題是由於 class 計數不匹配造成的。 您正在嘗試加載在不同 class 計數上訓練的 model 的重量。 您應該排除一些層，例如“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"])

請參閱此問題以獲取更多信息。