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] [CodeCamp #68] Add EfficientnetV2 Backbone #1253

Merged
merged 140 commits into from
Dec 30, 2022
Merged

[Feature] [CodeCamp #68] Add EfficientnetV2 Backbone #1253

merged 140 commits into from
Dec 30, 2022

Conversation

qingtian5
Copy link
Contributor

@qingtian5 qingtian5 commented Dec 9, 2022 •
edited by Ezra-Yu
Loading

Motivation

Add the EfficientnetV2 backbone and the convert weights.
paper: https://arxiv.org/abs/2104.00298
code: https://github.com/google/automl/tree/master/efficientnetv2

related to #1108 #1231

Modification

添加模型文件与配置文件

TODO

  • 对齐推理精度
    • 完成EfficientnetV2 框架搭设,
    • 对齐推理的transforms
    • 完成b0/b1/b2/b3架构设置，对齐推理精度
  • Code Review 以及重构代码
  • 完成单元测试以及Doc

训练精度对齐，不在这个PR中完成，但可以在对齐推理精度后测试
0. 跑一下源码和自己的配置，看下精度能否复现，(这里因为tf与torch的差异，可以跑timm的b0训练)

  1. 对齐训练transfroms
  2. 固定输入对齐前向loss
  3. 对齐反向梯度

Use cases (Optional)

If this PR introduces a new feature, it is better to list some use cases here and update the documentation.

Checklist

Before PR:

  • Pre-commit or other linting tools are used to fix the potential lint issues.
  • Bug fixes are fully covered by unit tests, the case that causes the bug should be added in the unit tests.
  • The modification is covered by complete unit tests. If not, please add more unit test to ensure the correctness.
  • The documentation has been modified accordingly, like docstring or example tutorials.

After PR:

  • If the modification has potential influence on downstream or other related projects, this PR should be tested with those projects, like MMDet or MMSeg.
  • CLA has been signed and all committers have signed the CLA in this PR.

qingtian4 and others added 27 commits December 5, 2022 15:03
@qingtian4
add efficientnet_v2.py
f18268c
@qingtian4
add efficientnet_v2 in __init__.py
6112044
@qingtian4
add efficientnet_v2_s base config file
c7d70c4
@qingtian4
add efficientnet_v2 config file
63364da
@qingtian4
add efficientnet_v2 config file
698a4ab
@qingtian4
update tuple output
ced8f39
@qingtian4
update config file
56ad534
@qingtian4
update model file
8ba74dd
@qingtian4
update model file
4b94f56
@qingtian4
update model file
317303a
@qingtian4
update config file
2c8df1b
@qingtian4
update model file
acbe09a
@qingtian4
Merge remote-tracking branch 'origin/dev-1.x' into dev-1.x
488e6d4
@qingtian4
update config file
4a4058c
@qingtian4
update model file
6ed2461
@qingtian4
update model file
6bc4119
@qingtian4
update model file
6dd9020
@qingtian4
update model file
9e8fb67
@qingtian4
update model file
3b6ae58
@qingtian4
update config file
8c0a996
@qingtian4
update config file
d398f16
@qingtian4
update model file
cd7d14c
@qingtian4
update model file
9414b17
@qingtian4
update model file
aaf7e5c
@qingtian4
update model file
ea81010
@qingtian4
update model config file
7302b63
@qingtian5
Update efficientnet_v2.py
c09d791
@CLAassistant
Copy link

CLAassistant commented Dec 9, 2022 •
edited
Loading

CLA assistant check
All committers have signed the CLA.

@Ezra-Yu Ezra-Yu changed the title CodeCamp #62 #EfficientnetV2 [WIP] [Feature] [CodeCamp #68] Add EfficientnetV2 Backbone Dec 9, 2022
Ezra-Yu
Ezra-Yu reviewed Dec 9, 2022
View reviewed changes
Copy link
Collaborator

@Ezra-Yu Ezra-Yu left a comment

Choose a reason for hiding this comment

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

  1. Please change to efficientnet-v2-s?
  2. add metafile and readme in configs/efficientnet_v2/

mmcls/models/backbones/efficientnet_v2.py Outdated Show resolved Hide resolved
mmcls/models/backbones/efficientnet_v2.py Outdated Show resolved Hide resolved
mmcls/models/backbones/efficientnet_v2.py Outdated Show resolved Hide resolved
mmcls/models/backbones/efficientnet_v2.py Outdated Show resolved Hide resolved
mmcls/models/backbones/efficientnet_v2.py Outdated Show resolved Hide resolved
mmcls/models/backbones/efficientnet_v2.py Outdated Show resolved Hide resolved
mmcls/models/backbones/efficientnet_v2.py Outdated Show resolved Hide resolved
mmcls/models/backbones/efficientnet_v2.py Outdated Show resolved Hide resolved
mmcls/models/backbones/efficientnet_v2.py Show resolved Hide resolved
mmcls/models/backbones/efficientnet_v2.py Outdated Show resolved Hide resolved
qingtian4 and others added 26 commits December 20, 2022 16:59
@qingtian4 @Ezra-Yu
update b0 and s train pipeline
3d4ce11
@qingtian4 @Ezra-Yu
update b0 and s train pipeline
140629b
@qingtian4 @Ezra-Yu
update b0 and s train pipeline
d3b86b0
@qingtian4 @Ezra-Yu
add test_efficientnet_v2
2f0e815
@qingtian4 @Ezra-Yu
update test_efficientnet_v2
e9d8dc5
@qingtian4 @Ezra-Yu
update model file docs
66a7c0e
@qingtian4 @Ezra-Yu
update test_efficientnet_v2
3740aca
@qingtian4 @Ezra-Yu
update test_efficientnet_v2
df162a2
@qingtian4 @Ezra-Yu
pass pre-commit hook
7683243
@Ezra-Yu
refactor efficientnetv2
5f6b1f1
@Ezra-Yu
refactor efficientnetv2
e64da5a
@Ezra-Yu
update readme, metafile and weight links
8e84ba3
@Ezra-Yu
update model-index.yml
31b8f2c
@Ezra-Yu
fix lint
72a6644
@Ezra-Yu
fix typo
dc13b91
@qingtian5
Update efficientnetv2-b1_8xb32_in1k.py
b4feb11
@qingtian5
Update efficientnetv2-b2_8xb32_in1k.py
d7a080b
@qingtian5
Update efficientnetv2-b3_8xb32_in1k.py
6587c37
@qingtian4
Merge branch 'dev-1.x' of https://github.com/suibe-qingtian/mmclassif…
af720f5 
…ication into dev-1.x

� Conflicts:
�	configs/efficientnet_v2/README.md
�	configs/efficientnet_v2/metafile.yml
�	mmcls/models/backbones/efficientnet_v2.py
�	tests/test_models/test_backbones/test_efficientnet_v2.py
�	tools/model_converters/efficientnetv2_to_mmcls.py
@qingtian4
update two moduals and model file
ec48947
@qingtian4
update modual file
b3a572f
@Ezra-Yu
update accuracys
1068d74
@Ezra-Yu
update accuracys
86f8c9a
@Ezra-Yu
update metafile
f0733ea
@Ezra-Yu
fix build docs
a423ecf
@Ezra-Yu
update links
be419f7
Ezra-Yu
Ezra-Yu approved these changes Dec 21, 2022
View reviewed changes
Copy link
Collaborator

@Ezra-Yu Ezra-Yu left a comment

Choose a reason for hiding this comment

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

LGTM.

@mzr1996 mzr1996 merged commit 74743ef into open-mmlab:dev-1.x Dec 30, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@Ezra-Yu Ezra-Yu Ezra-Yu approved these changes

@mzr1996 mzr1996 mzr1996 approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
5 participants
@qingtian5 @CLAassistant @Ezra-Yu @mzr1996 @qingtian4