Skip to content

4️⃣ Use a Stronger Pretrained Backbone #4

@Microcosmos22

Description

@Microcosmos22

Your MobileNet-based custom backbone has no ImageNet weights right now:

torchvision.models.mobilenet_v3_small(pretrained=False)

Switch to:

torchvision.models.mobilenet_v3_small(weights="DEFAULT")

or even better:
💪 resnet50_fpn backbone (default for Mask R-CNN)
→ larger, pretrained feature extractor → huge boost.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions