Detectron for PyTorch (work in progress)

Hi,

I’m working on using detectron models with PyTorch. Inference and evaluation are working. Caffe2 models can be used.

In case you are interested it’s here.

It’d be nice if some people contribute to expand it a bit more. Main missing features are:
-Losses to be able to train
-FPN

3 Likes