python - 使用 tensorflow-distributed 实现 mask-r-cnn

Question

我正在训练 mask-r-cnn network ，它建立在 tensorflow 和 keras 之上。我正在寻找一种减少训练时间的方法，所以我想用 tensorflow-distributed 来实现它。

我使用 mask-r-cnn 已经有一段时间了，但似乎我正在尝试做的事情需要我修改 mask-r-cnn 的源代码，这超出了我目前的技能。

所以，我的问题是，有人做过吗，或者类似的事情？有可能吗，还是我误解了 tensorflow-distributed 的使用。

提前致谢。

Answer 1

即使没有分布式训练，Mask R-CNN 在tensorflow 中的tensorpack 实现runs 5x faster (也比您链接到的那个更准确)。

它还支持使用 MPI 进行分布式训练。