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

Add MobileNetV3 architecture for Detection #3253

Merged
merged 12 commits into from
Jan 18, 2021

Conversation

datumbox
Copy link
Contributor

@datumbox datumbox commented Jan 14, 2021

Fixes #1999

Trained using the code committed at 11408d6.

The current temporary pre-trained model was trained:

python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\
    --dataset coco --model fasterrcnn_mobilenet_v3_large_fpn --epochs 26\
    --lr-steps 16 22 --aspect-ratio-group-factor 3

Submitted batch job 34945231

Accuracy metrics:

0: IoU metric: bbox
0:  Average Precision  (AP) @[ IoU=0.50:0.95 | area=   all | maxDets=100 ] = 0.230
0:  Average Precision  (AP) @[ IoU=0.50      | area=   all | maxDets=100 ] = 0.385
0:  Average Precision  (AP) @[ IoU=0.75      | area=   all | maxDets=100 ] = 0.233
0:  Average Precision  (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.026
0:  Average Precision  (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.223
0:  Average Precision  (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.446
0:  Average Recall     (AR) @[ IoU=0.50:0.95 | area=   all | maxDets=  1 ] = 0.218
0:  Average Recall     (AR) @[ IoU=0.50:0.95 | area=   all | maxDets= 10 ] = 0.298
0:  Average Recall     (AR) @[ IoU=0.50:0.95 | area=   all | maxDets=100 ] = 0.305
0:  Average Recall     (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.041
0:  Average Recall     (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.315
0:  Average Recall     (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.577

Validated with:

python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\
 --model fasterrcnn_mobilenet_v3_large_fpn --test-only --pretrained

Submitted batch job 34643680

Speed benchmark:
0.14 sec per image on CPU

@datumbox datumbox force-pushed the mobilenetv3_detection branch from 24f1e90 to 30299cc Compare January 14, 2021 16:40
@datumbox datumbox mentioned this pull request Jan 14, 2021
13 tasks
@datumbox datumbox force-pushed the mobilenetv3_detection branch 2 times, most recently from 0455f35 to 30299cc Compare January 15, 2021 09:29
@datumbox datumbox force-pushed the mobilenetv3_detection branch from 4609c36 to 11408d6 Compare January 17, 2021 22:51
@datumbox datumbox changed the title [WIP] Add MobileNetV3 architecture for Detection Add MobileNetV3 architecture for Detection Jan 18, 2021
@datumbox datumbox requested a review from fmassa January 18, 2021 13:08
Copy link
Member

@fmassa fmassa left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks great to me, thanks a lot Vasilis!

@fmassa fmassa merged commit bf211da into pytorch:master Jan 18, 2021
@datumbox datumbox deleted the mobilenetv3_detection branch January 18, 2021 19:06
facebook-github-bot pushed a commit that referenced this pull request Jan 21, 2021
Summary:
* Minor refactoring of a private method to make it reusuable.

* Adding a FasterRCNN + MobileNetV3 with & w/o FPN models.

* Reducing Resolution to 320-640 and anchor sizes to 16-256.

* Increase anchor sizes.

* Adding rpn score threshold param on the train script.

* Adding trainable_backbone_layers param on the train script.

* Adding rpn_score_thresh param directly in fasterrcnn_mobilenet_v3_large_fpn.

* Remove fasterrcnn_mobilenet_v3_large prototype and update expected file.

* Update documentation and adding weights.

* Use buildin Identity.

* Fix spelling.

Reviewed By: datumbox

Differential Revision: D25954553

fbshipit-source-id: 41009b5c6e855cd0b2639902165c57402387d0be
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Request Mobilenet fpn
3 participants