Como o Caffe2 agora foi movido para o repositório pytorch como back-end, há algum plano para migrar o Detectron para o Pytorch? :)
Você pode acompanhar o "merge" aqui https://github.com/caffe2/caffe2/issues/2439.
Caro @gadcam ,
Você pode consultar a versão PyTorch do Detectron .
Eu fiz uma implementação pytorch que poderia fazer treinamento em multi-gpus.
Estaremos analisando como portar o código para usar o frontend pytorch à medida que a fusão caffe2 e pytorch progride.
@rbgirshick Obrigado por seus grandes esforços! Algum progresso sobre a interface do Pytorch agora?
Eu recomendo dar uma olhada em:
https://github.com/facebookresearch/maskrcnn-benchmark
https://github.com/open-mmlab/mmdetection
Eles ainda não estão completos em comparação com este repositório, mas são ótimos projetos.
@rbgirshick Algum progresso na portabilidade do Detectron para o pytorch?
Comentários muito úteis
Estaremos analisando como portar o código para usar o frontend pytorch à medida que a fusão caffe2 e pytorch progride.