Detectron: Algum plano para migrar o detectron para o pytorch?

Criado em 2 abr. 2018  ·  7Comentários  ·  Fonte: facebookresearch/Detectron

Como o Caffe2 agora foi movido para o repositório pytorch como back-end, há algum plano para migrar o 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.

Todos 7 comentários

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?

Esta página foi útil?
0 / 5 - 0 avaliações