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

Question

我正在训练一个基于 tensorflow 和 keras的mask-r-cnn 网络。我正在寻找一种减少训练时间的方法，所以我想用 tensorflow-distributed 来实现它。

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

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

提前谢谢。

Answer 1

即使没有分布式训练，在 tensorflow 中实现 Mask R-CNN 的 tensorpack运行速度也比您链接的那个快 5 倍（也更准确）。

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