Pre-trained detection models for PyTorch Hub

🚀 Feature

First of all, thanks for the amazing work.

I noticed currently there’s no pre-trained detection models exposed via hub_conf.py. It looks like the building blocks are already there. Is there plan to support it?

Thanks

Motivation

Increase coverage of supported vision tasks via Torch Hub.

Pitch

• entrypoints for detection models are added to hub_conf.py
• some helpers e.g. for bbox parsing could also be added there

[NicolasHug]

As @datumbox mentioned in general one should not expect models with compiled extensions to work with torchhub.

However, since you already have torchvision 0.11 installed, these extensions have been compiled already and are present somewhere on your machine.

import torch and import torchvision do quite a bit of magic under the hood, resolving dynamic lib symbols, and all sorts of shenanigans that I don’t understand enough. But I would bet that when you’re on the 0.11 branch, those symbols get resolved properly, which is why you’re able to load the models in this specific case (and why it still fails if import torchvision is removed)

[datumbox]

I believe what you describe is expected. The specific version that you have installed will work because the binaries contain the compiled C++/CUDA ops.

I’m not an expert on the internals of TorchHub. @NicolasHug do you know if the point 2 on the Observation section above is expected?