Detectron: Any plan to migrate detectron to pytorch?

Created on 2 Apr 2018  ·  7Comments  ·  Source: facebookresearch/Detectron

Since Caffe2 is now moved to the pytorch repo as its backend, is there any plan to migrate Detectron to Pytorch? : )

Most helpful comment

We will be looking into porting the code the use the pytorch frontend as caffe2 and pytorch merging progresses.

All 7 comments

You can follow the "merge" here https://github.com/caffe2/caffe2/issues/2439.

Dear @gadcam,
You can refer to PyTorch version of Detectron.

I have made a pytorch implementation that could do training on multi-gpus.

We will be looking into porting the code the use the pytorch frontend as caffe2 and pytorch merging progresses.

@rbgirshick Thank you for your great efforts! Any progress about the Pytorch frontend now?

I recommend taking a look at:

https://github.com/facebookresearch/maskrcnn-benchmark
https://github.com/open-mmlab/mmdetection

They are not yet feature complete compared to this repo, but they are great projects.

@rbgirshick Any further progress on porting Detectron to pytorch?

Was this page helpful?
0 / 5 - 0 ratings

Related issues

elfpattern picture elfpattern  ·  3Comments

fangpengcheng95 picture fangpengcheng95  ·  4Comments

gaopeng-eugene picture gaopeng-eugene  ·  4Comments

Hwang-dae-won picture Hwang-dae-won  ·  3Comments

kampelmuehler picture kampelmuehler  ·  4Comments