Étant donné que Caffe2 est maintenant déplacé vers le référentiel pytorch en tant que backend, est-il prévu de migrer Detectron vers Pytorch ? : )
Vous pouvez suivre la "fusion" ici https://github.com/caffe2/caffe2/issues/2439.
Cher @gadcam ,
Vous pouvez vous référer à la version PyTorch de Detectron .
J'ai fait une implémentation pytorch qui pourrait faire une formation sur multi-gpus.
Nous examinerons le portage du code et l'utilisation de l'interface pytorch au fur et à mesure que la fusion de caffe2 et pytorch progresse.
@rbgirshick Merci pour vos efforts considérables ! Des progrès sur le frontend Pytorch maintenant?
Je vous conseille de jeter un œil à :
https://github.com/facebookresearch/maskrcnn-benchmark
https://github.com/open-mmlab/mmdetection
Ils ne sont pas encore complets par rapport à ce référentiel, mais ce sont de grands projets.
@rbgirshick De nouveaux progrès sur le portage de Detectron vers pytorch ?
Commentaire le plus utile
Nous examinerons le portage du code et l'utilisation de l'interface pytorch au fur et à mesure que la fusion de caffe2 et pytorch progresse.