12 Commits

Author SHA1 Message Date
lostkevin
31897984d8
implement onnx export for inception3/4, resnext, mobilenetv2 (#346)
* add inceptionv4 backbone/training settings
* add converted backbone, top-1 acc 80.08
2024-07-18 16:52:56 +08:00
gulou
8c90ceaf84
add resnet export of onnx (#341)
* add checkpoint_sync_export for resnet config
2024-07-02 19:39:28 +08:00
tuofeilun
23f2b0e399
Adapt designer (#235)
1. Use original config as startup script. (For details, see refactor config parsing method #225)
2. Refactor the splicing rules of the check_base_cfg_path function in the EasyCV/easycv/utils/config_tools.py
3. Support three ways to pass class_list parameter.
4. Fix the bug that clsevalutor may make mistakes when evaluating top5.
5. Fix the bug that the distributed export cannot export the model.
6. Fix the bug that the load pretrained model key does not match.
7. support cls data source itag.
2022-12-01 17:47:10 +08:00
Cathy0908
5ac6381758
add error code (#146)
* add error code
2022-09-19 16:07:04 +08:00
zzoneee
0cb91de0cb
add DeiT III (#171)
1.Add a backbone: deitiii.
2.Add an optimizer: lamb. 
3.Add a sampler: RASampler. 
4.Add a lr update hook: CosineAnnealingWarmupByEpochLrUpdaterHook.
5.In easycv/models/classification/classification.py, I remove the default mixup_cfg to keep the classification.py clean.
2022-09-14 15:24:54 +08:00
Chen Jiayu
9517bb80ff
add dn_detr algo (#119)
support dn-detr-r50: 44.39
refactor set_criterion
2022-07-27 15:06:06 +08:00
Cathy0908
b737027aa4
remove useless import and optimize imports (#84) 2022-06-13 10:29:41 +08:00
Chen Jiayu
12e3bed42b
bugfix_swintiny (#79)
1. add easycv_model default_path
2. bugfix pretrained
2022-06-06 12:03:05 +08:00
Chen Jiayu
3ee118f065
Expand cls model zoo (#55)
1. expand cls model zoo
2. uniform load_pretrained
2022-06-01 11:01:29 +08:00
Cathy0908
13ff8263b3
remove all imports in easycv.utils.__init__ (#21) 2022-04-22 15:22:43 +08:00
jiangnana.jnn
e8008e3abd [Feature]: uniform input and output format for transforms 2022-04-06 20:02:21 +08:00
wenmeng.zwm
055deba53e initial commit 2022-04-02 20:01:06 +08:00