由于 Caffe2 现在移至 pytorch repo 作为其后端,是否有将 Detectron 迁移到 Pytorch 的计划? :)
亲爱的@gadcam ,
你可以参考Detectron 的 PyTorch 版本。
我已经制作了一个可以在多 GPU 上进行培训的pytorch 实现。
我们将研究移植使用 pytorch 前端的代码作为 caffe2 和 pytorch 合并的进展。
@rbgirshick感谢您的巨大努力! Pytorch 前端现在有什么进展吗?
我建议看一下:
https://github.com/facebookresearch/maskrcnn-benchmark
https://github.com/open-mmlab/mmdetection
与这个 repo 相比,它们的功能还不完整,但它们是很棒的项目。
@rbgirshick在将 Detectron 移植到 pytorch 方面有什么进一步的进展吗?
最有用的评论
我们将研究移植使用 pytorch 前端的代码作为 caffe2 和 pytorch 合并的进展。