Skip to content

[OTX/Semi-SL-Det] Enable training and evaluation for unbiased teacher structure#1398

Merged
jaegukhyun merged 23 commits intofeature/otxfrom
feature/otx-semisl-det
Dec 23, 2022
Merged

[OTX/Semi-SL-Det] Enable training and evaluation for unbiased teacher structure#1398
jaegukhyun merged 23 commits intofeature/otxfrom
feature/otx-semisl-det

Conversation

@jaegukhyun
Copy link
Copy Markdown
Contributor

@jaegukhyun jaegukhyun commented Nov 28, 2022

This PR is for enabling training and evaluation for unbiased teacher structure.

'semisl' folder will be placed inside of each detection model config folder. 'semisl' folder would have template_semi.yaml, model.py, data_pipeline.py for semi-supervised learning.
Detection task determine selects its training algorithm below logic:
if template.TrainType == "SEMISUPERVISED" and 'unlabeled' in cfg.data_cfg:
train_type = "SEMISUPERVISED"
else:
train_type = "INCREMENTAL"
In addition detection task transfer its training algorithm through setting self._recipe_cfg.train_type

@jaegukhyun jaegukhyun marked this pull request as ready for review December 7, 2022 06:08
@jaegukhyun jaegukhyun requested a review from a team as a code owner December 7, 2022 06:08
@jaegukhyun
Copy link
Copy Markdown
Contributor Author

There are two major changes

  1. Change template directory structure to one template for one model policy
  2. Change test method using configurable training type

@jaegukhyun jaegukhyun requested review from JihwanEom and harimkang and removed request for eugene123tw, kprokofi, sovrasov, sungmanc and supersoob December 16, 2022 02:13
Copy link
Copy Markdown
Contributor

@harimkang harimkang left a comment

Choose a reason for hiding this comment

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

It looks good. I left some concerned comments.

harimkang
harimkang previously approved these changes Dec 16, 2022
Copy link
Copy Markdown
Contributor

@harimkang harimkang left a comment

Choose a reason for hiding this comment

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

LGTM. Let's wait for the CI to be fixed and check the tests.

Copy link
Copy Markdown
Contributor

@sungmanc sungmanc left a comment

Choose a reason for hiding this comment

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

I left some comments please check it

supersoob
supersoob previously approved these changes Dec 19, 2022
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, thanks for the hard work. I left one command, but just suggestion.

supersoob
supersoob previously approved these changes Dec 20, 2022
@github-actions github-actions bot added CLI Any changes in OTE CLI ALGO Any changes in OTX Algo Tasks implementation labels Dec 21, 2022
@jaegukhyun
Copy link
Copy Markdown
Contributor Author

https://github.com/openvinotoolkit/training_extensions/actions/runs/3754946842/jobs/6380632103
@harimkang, @supersoob, @sungmanc Tests are passed except 3 openvino eval tests. The failed cases are from performance difference after openvino export, and they are not related with this PR. I think we can finish this PR

Copy link
Copy Markdown
Contributor

@sungmanc sungmanc 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
Contributor

@harimkang harimkang 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 8299edf into feature/otx Dec 23, 2022
@jaegukhyun jaegukhyun deleted the feature/otx-semisl-det branch December 23, 2022 00:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ALGO Any changes in OTX Algo Tasks implementation CLI Any changes in OTE CLI TEST Any changes in tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants