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.

Sign up for GitHub

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

Jump to bottom

[Feature] Support STDC Network (new) #995

Merged
merged 39 commits into from
Dec 10, 2021
Merged

[Feature] Support STDC Network (new) #995

merged 39 commits into from
Dec 10, 2021

Conversation

MengzhangLI
Copy link
Contributor

@MengzhangLI MengzhangLI commented Oct 25, 2021 •
edited
Loading

The old PR: #865

Motivation

Support a new real-time algorithm, Rethinking BiSeNet For Real-time Semantic Segmentation (CVPR 2021).

Modification

  1. Add a new backbone modified from https://github.com/MichaelFan01/STDC-Seg
  2. Add config files.

Update(2021-10-25):

  • Aligning inference metric.
  • Add Unit Test.
  • Aligning training metric.

The inference metrics of STDC1-Seg50(72.17) and STDC2-Seg50 (74.16) have been aligned.

STDC1-Seg50 (72.17)

image

STDC2-Seg50 (74.16)
image

Update (2021-12-08)

  • Refactor STDCContextPathNet forward.
  • Refactor hacky code of stdc2mmseg.py
  • Upload models (after transferring keys) and logs.

@MengzhangLI MengzhangLI self-assigned this Oct 25, 2021
@MengzhangLI MengzhangLI added the WIP Work in process label Oct 25, 2021
@MengzhangLI MengzhangLI mentioned this pull request Nov 1, 2021
Open
@RockeyCoss
Copy link
Contributor

Please merge the master branch into your branch, thank you.

@MengzhangLI MengzhangLI added the Merging PR waited for merging label Dec 8, 2021
@MengzhangLI
Copy link
Contributor Author

MengzhangLI commented Dec 9, 2021 •
edited
Loading

Inference metric of new models (after refactory of stdc.py and transferring keys of .pth) have been tested.

STDC1:
image

STDC1 (pretrained):
image

STDC2:
image

STDC2 (pretrained):
image

xvjiarui
xvjiarui approved these changes Dec 9, 2021
View reviewed changes
Copy link
Collaborator

@xvjiarui xvjiarui left a comment

Choose a reason for hiding this comment

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

Just a few minor comments

@MengzhangLI MengzhangLI removed Merging PR waited for merging WIP Work in process labels Dec 10, 2021
Junjun2016
Junjun2016 approved these changes Dec 10, 2021
View reviewed changes
Copy link
Collaborator

@Junjun2016 Junjun2016 left a comment

Choose a reason for hiding this comment

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

LGTM

@Junjun2016 Junjun2016 merged commit 5783bc1 into open-mmlab:master Dec 10, 2021
@MengzhangLI MengzhangLI mentioned this pull request Feb 18, 2022
Closed
bowenroom pushed a commit to bowenroom/mmsegmentation that referenced this pull request Feb 25, 2022
@MengzhangLI
[Feature] Support STDC Network (new) (open-mmlab#995)
c1e7251 
* refactor stdc code

* update key

* fix backbone inference

* remove comments

* fixing errors

* fixing version conflict

* fux typo

* use STDCHead

* upload models&logs

* adding model converters script and fix unittest

* fix error

* fix error

* fix error

* delete redundant keys in config

* fix errors in configs and unittest

* fix errors in configs and unittest

* fix errors in configs and unittest

* change Memory name

* refactor stdc2mmseg

* change name to STDC

* refactor stdc

* refactor stdc

* stdc refactor

* stdc refactor

* stdc refactor

* stdc refactor

* stdc refactor

* stdc refactor

* refactor stdc

* stdc refactor

Co-authored-by: xiexinch <xinchen.xie@qq.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@xvjiarui xvjiarui xvjiarui approved these changes

@Junjun2016 Junjun2016 Junjun2016 approved these changes

Assignees

@MengzhangLI MengzhangLI

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
4 participants
@MengzhangLI @RockeyCoss @Junjun2016 @xvjiarui