Dado que Caffe2 ahora se mueve al repositorio de pytorch como su backend, ¿hay algún plan para migrar Detectron a Pytorch? :)
Puede seguir la "fusión" aquí https://github.com/caffe2/caffe2/issues/2439.
Estimado @gadcam ,
Puede consultar la versión PyTorch de Detectron .
Hice una implementación de pytorch que podría hacer entrenamiento en multi-gpus.
Buscaremos portar el código y usar la interfaz de pytorch a medida que avanza la fusión de caffe2 y pytorch.
@rbgirshick ¡ Gracias por sus grandes esfuerzos! ¿Algún progreso sobre la interfaz de Pytorch ahora?
Recomiendo echar un vistazo a:
https://github.com/facebookresearch/maskrcnn-benchmark
https://github.com/open-mmlab/mmdetection
Aún no tienen características completas en comparación con este repositorio, pero son grandes proyectos.
@rbgirshick ¿ Algún progreso adicional en la portabilidad de Detectron a pytorch?
Comentario más útil
Buscaremos portar el código y usar la interfaz de pytorch a medida que avanza la fusión de caffe2 y pytorch.