search

Lightning-Universe / lightning-bolts

Toolbox of models, callbacks, and datasets for AI/ML researchers.
https://lightning-bolts.readthedocs.io
Apache License 2.0
1.69k stars 323 forks source link

Add Mobilenet V3 Backbones to Detection #528

Open oke-aditya opened 3 years ago

oke-aditya commented 3 years ago

🚀 Feature

Next release of torchvision packs mobilenet v3 and mobilenet v3 FPN backbone. We can add that to supported backbones for Faster RCNN and RetinaNet

Motivation

We need backbones, they enhance detection

Pitch

Same way add backbones to pl_bolts.models.detection.componenets as done earlier.

But wait for Torchvision next release 0.9.

Additional context

I will complete this once torchvision 0.9 is released (or do we test with torchvsion master ?)

akihironitta commented 3 years ago

@oke-aditya Sounds nice!