Skip to content
This repository was archived by the owner on Apr 17, 2023. It is now read-only.

[Semi-SL/Detection] Refactoring mpa/det/ to support multiple learning technique#94

Merged
jaegukhyun merged 15 commits intootxfrom
semisl-det
Dec 7, 2022
Merged

[Semi-SL/Detection] Refactoring mpa/det/ to support multiple learning technique#94
jaegukhyun merged 15 commits intootxfrom
semisl-det

Conversation

@jaegukhyun
Copy link
Copy Markdown
Contributor

@jaegukhyun jaegukhyun commented Nov 28, 2022

This PR includes refactoring mpa/det codes.

Current mpa/det's codes highly focus on class incremental learning, and this make hard to enable other training method.
To solve this issue, I split configure function in mpa/det/stage.py into two things. First is configuring configs for general fine tuning. Second is configuring configs for class-incremental learning.
And I created ClassIncrStage, which inherit base Stage, to support class incremental learning. Now trainer at mpa/det inherits from ClassIncrStage, because class incremental learning is our default recipe.
To enable semi-sl for detection, SemiStage has to be different with ClassIncrStage. Therefore I created SemiStage. SemiStage inherits from ClassIncrStage, and it modifies some configuration function for supporting semi-sl. Since it inherits ClassIncrStage, it can support class incremental scenarios.
As SemiStage is newly generated, Trainer and Inferencer for semi-sl is also generated, and they inherits SemiStage.

You can test this with otx branch for semi-sl(https://github.com/openvinotoolkit/training_extensions/tree/feature/otx-semisl-det). From this PR link(open-edge-platform/training_extensions#1398), you can find a description for this otx branch

@jaegukhyun
Copy link
Copy Markdown
Contributor Author

@supersoob @sovrasov @kprokofi @JihwanEom It's time to merge this PR, please review this PR. I want you to add your implementation for semi-sl from this branch. After merging this PR, I will open OTX PR, because semi-sl cli test requires this mpa branch

Copy link
Copy Markdown

@supersoob supersoob left a comment

Choose a reason for hiding this comment

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

Overall, it looks good to me. Let's wait for other's reviews. After merging this PR, I think it's gonna be good to make each PR for each task, referring this PR's code change. Thanks for your hard work, Jaeguk:)

Copy link
Copy Markdown

@goodsong81 goodsong81 left a comment

Choose a reason for hiding this comment

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

I support @supersoob's point.
Why don't you create classincr directory?
Being a default option has nothing to do with class/directory structure.
Anyway, it's just a suggestion.

Copy link
Copy Markdown
Contributor

@JihwanEom JihwanEom left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Copy Markdown

@supersoob supersoob left a comment

Choose a reason for hiding this comment

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

LGTM

@jaegukhyun jaegukhyun merged commit d6d3e51 into otx Dec 7, 2022
@jaegukhyun jaegukhyun deleted the semisl-det branch December 7, 2022 06:05
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants