时间戳

首页 > 解决方案 > TensorFlow 2 Mask-RCNN？

docker - TensorFlow 2 Mask-RCNN？

问题描述

我似乎找不到适用于 TensorFlow 2 的可靠版本的 Mask-RCNN。matterport mask-rcnn ( https://github.com/matterport/Mask_RCNN ) 已经贬低了 TensorFlow 1 代码。有人知道 RCNN 或其他对象检测模型的 TensorFlow 2 实现吗？或者可能是一个可以与 Matterport 一起使用的稳定 Docker 镜像？

标签： dockertensorflowmachine-learningneural-networkfaster-rcnn

解决方案

在 TF 2+ 中有一个 Mask RCNN 的工作实现。我在这里找到它：https ://github.com/matterport/Mask_RCNN/tree/295c802f0bebbc4a34ec4855f4960a52a701271d

要使所有示例在 TF 2.4 中工作，您必须通过替换修改 rcnn/model.py 文件：

if model.uses_learning_phase and not isinstance(K.learning_phase(), int):

和

if not isinstance(K.learning_phase(), int):

因为 uses_learning_phase 在 TF 2.4 中不再起作用。

然后使用此更正：https ://stackoverflow.com/a/66842506/13467454

它应该在 TF 2.4 上运行良好。

推荐阅读