Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Mask R-CNN with MobileNet v3 backbone #4048

Open
niladridutt opened this issue Jun 11, 2021 · 3 comments
Open

Mask R-CNN with MobileNet v3 backbone #4048

niladridutt opened this issue Jun 11, 2021 · 3 comments

Comments

@niladridutt
Copy link

🚀 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

  • Similar module like torchvision.models.detection.fasterrcnn_mobilenet_v3_large_fpn but to work with Mask R-CNN like resnet has torchvision.models.detection.maskrcnn_resnet50_fpn
  • MobileNet V3 would make inference much faster
  • Mask R-CNN is widely used owing to its dual use of getting mask+bounding box
@datumbox
Copy link
Contributor

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

@niladridutt
Copy link
Author

Thanks!

@oke-aditya
Copy link
Contributor

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants