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 YOLACT support from mmdet #1046

Merged
merged 3 commits into from
Jan 26, 2022
Merged

Conversation

dnth
Copy link
Contributor

@dnth dnth commented Jan 25, 2022

This PR adds YOLACT: Real-time Instance Segmentation, a single stage instance segmentation model, into IceVision.

Paper: https://arxiv.org/abs/1904.02689
Configs/models: https://github.com/open-mmlab/mmdetection/blob/master/configs/yolact/README.md

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@codecov
Copy link

codecov bot commented Jan 25, 2022

Codecov Report

Merging #1046 (6f4658e) into master (bfbed41) will decrease coverage by 0.03%.
The diff coverage is 80.48%.

Impacted file tree graph

@@            Coverage Diff             @@
##           master    #1046      +/-   ##
==========================================
- Coverage   85.96%   85.92%   -0.04%     
==========================================
  Files         272      277       +5     
  Lines        5794     5835      +41     
==========================================
+ Hits         4981     5014      +33     
- Misses        813      821       +8     
Flag Coverage Δ
unittests 85.92% <80.48%> (-0.04%) ⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files Coverage Δ
icevision/models/mmdet/utils.py 57.81% <0.00%> (-3.86%) ⬇️
icevision/models/mmdet/common/utils.py 81.81% <25.00%> (-5.69%) ⬇️
...ion/models/mmdet/common/mask/single_stage/model.py 90.90% <90.90%> (ø)
.../models/mmdet/common/mask/single_stage/__init__.py 100.00% <100.00%> (ø)
icevision/models/mmdet/models/__init__.py 100.00% <100.00%> (ø)
icevision/models/mmdet/models/yolact/__init__.py 100.00% <100.00%> (ø)
...n/models/mmdet/models/yolact/backbones/__init__.py 100.00% <100.00%> (ø)
...models/mmdet/models/yolact/backbones/resnet_fpn.py 100.00% <100.00%> (ø)

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update bfbed41...6f4658e. Read the comment docs.

@ai-fast-track
Copy link
Collaborator

Thanks a lot @dnth!
Did you test it in the getting_started_instance_segmentation notebook?
What's the COCO metric score?

@dnth
Copy link
Contributor Author

dnth commented Jan 26, 2022

Thanks a lot @dnth! Did you test it in the getting_started_instance_segmentation notebook? What's the COCO metric score?

Yes! I tested the model with the getting started notebook on my machine. The highest score was 0.76 as shown in the screenshot.

Screenshot from 2022-01-26 08-10-46

I have also tested on Colab. Here is the link to the notebook.
https://colab.research.google.com/drive/10PUIfqvI5R9dydmn15qPTVNFdhLHhChR?usp=sharing

@ai-fast-track ai-fast-track merged commit eb550f8 into airctic:master Jan 26, 2022
@dnth dnth deleted the yolact-support branch April 5, 2022 05:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants