pytorch / vision

Datasets, Transforms and Models specific to Computer Vision
https://pytorch.org/vision
BSD 3-Clause "New" or "Revised" License
16.29k stars 6.96k forks source link

Mask R-CNN with MobileNet v3 backbone #4048

Open niladridutt opened 3 years ago

niladridutt commented 3 years ago

🚀 Feature

Add Mask R-CNN with MobileNet v3 backbone

Motivation

In the recent release torchvision added support for mobilenet v3 and extended it to work with instance segmentation models. torchvision.models.detection.fasterrcnn_mobilenet_v3_large_fpn However, support for Mask R-CNN with MobileNet v3 backbone is still missing and would be a worthy addition.

Pitch

datumbox commented 3 years ago

@niladridutt Thanks, makes sense. We can have a look at it.

niladridutt commented 3 years ago

Thanks!

oke-aditya commented 3 years ago

Let's add this to #2707 as well :)