mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
262 Commits
8 Branches
41 Tags
32 MiB
Commit Graph

262 Commits (b30f79ea4ceb855d1a5540673d1951a3b981bf57)
 

Author SHA1 Message Date
whcao b30f79ea4c
[Feature]Modify Parameters Passing in models.heads (#239) 
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* set cal_acc in ClsHead defaults to False

* set cal_acc defaults to False

* use *args, **kwargs instead

* change bs16 to 3 in test_image_classifier_vit

* fix some comments

* change cal_acc=True

* test LinearClsHead
 2021-05-10 14:56:55 +08:00
LXXXXR 37167158e7
bump version to v0.11.0 (#233) 2021-05-01 22:26:39 +08:00
LXXXXR 761c95925d
support mmcli (#232) 2021-05-01 22:24:16 +08:00
QingChuanWS f970a36898
update pytorch2onnx.md table (#229) 2021-04-30 17:18:06 +08:00
whcao 3be95b99c2
[Feature]Modify modelzoo readme (#230) 
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* add imagenet_bs4096_AdamW.py

* delete 2 lines of comments

* change bs to 64

* fix bug

* add vit to model_zoo.md

* rename
 2021-04-29 15:18:55 +08:00
whcao 16947f1239
[Bug]Fix weight decay (#227) 
* add imagenet bs 4096

* add vit_base_patch16_224_finetune

* add vit_base_patch16_224_pretrain

* add vit_base_patch16_384_finetune

* add vit_base_patch16_384_finetune

* add vit_b_p16_224_finetune_imagenet

* add vit_b_p16_224_pretrain_imagenet

* add vit_b_p16_384_finetune_imagenet

* add vit

* add vit

* add vit head

* vit unitest

* keep up with ClsHead

* test vit

* add flag to determiine whether to calculate acc during training

* Changes related to mmcv1.3.0

* change checkpoint saving interval to 10

* add label smooth

* default_runtime.py recovery

* docformatter

* docformatter

* delete 2 lines of comments

* delete configs/_base_/schedules/imagenet_bs4096.py

* add configs/_base_/schedules/imagenet_bs2048_AdamW.py

* rename imagenet_bs4096.py to imagenet_bs2048_AdamW.py

* add AutoAugment

* fix weight decay in vit

* change eval interval to 10

* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* delete @torch.jit.ignore

* change eval interval back to 1

* add some comments to imagenet_bs2048_AdamW

* add some comments
 2021-04-28 17:16:43 +08:00
LXXXXR 5fb9ced5f9
[Docs] Change tag into comment (#226) 
* change tags

* remove modelzoo_statistics
 2021-04-28 17:08:57 +08:00
LXXXXR a604cd97ef
[Feature] Adding PWC metafile (#225) 
* add yml

* fix VGG

* minor fix

* add model zoo

* add bs and epoch

* revised according to comments

* revised according to comments

* revised according to comments
 2021-04-28 17:07:26 +08:00
mzr1996 3d87474c71
Add Chinese Readme and docs (#221) 
* Add Citation in README

Add Citation and mmgeneration in README

* Merge inference and test section in getting_start.md and other small chagne.

* Fix code type in install.md

* Add Chinese Readme

* README and docs improvement.
 2021-04-26 13:58:18 +08:00
QingChuanWS 01d2849b76
[Feature]: add onnxruntime test tool (#212) 
* [draft] add onnxruntime accuruacy verification

* fix a bug

* update code

* fix lint

* fix lint

* update code and doc

* update doc

* update code

* update code

* updata doc and updata code

* update doc and fix some bug

* update doc

* update doc

* update doc

* update doc

* update doc

* update doc

* fix bug

* update doc

* update code

* move CUDAExecutionProvider to first place

* update resnext accuracy

* update doc

Co-authored-by: maningsheng <maningsheng@sensetime.com>
 2021-04-26 13:57:08 +08:00
WRH 9be435846c
Support training on CPU (#219) 
* draft

* add parameter for training tools

* Update .pre-commit-config.yaml
 2021-04-26 13:56:45 +08:00
LXXXXR d72d233f9b
fix resnext resize (#222) 2021-04-25 16:26:34 +08:00
whcao 759aa92437
[Bug]add cutmix_prob to config (#220) 
* add cutmix_prob to config

* empty
 2021-04-21 21:57:09 +08:00
whcao 31a6a362ba
Add some vit configs (#217) 
* add vit_base_patch32_384_finetune.py

* add vit_base_patch32_384_finetune_imagenet.py to vision_transformer

* add vit_large_patch16_384_finetune.py to models

* add vit_large_patch16_384_finetune_imagenet.py to vision_transformer

* add vit_large_patch32_384_finetune to models

* add vit_large_patch32_384_finetune_imagenet to vision_transformer

* add vit_large_patch16_224_finetune.py to models

* add vit_large_patch16_224_finetune_imagenet.py to vision_transformer

* delete some useless comments
 2021-04-20 11:32:20 +08:00
whcao affb39fe07
[Feature]Add Vit (#214) 
* add imagenet bs 4096

* add vit_base_patch16_224_finetune

* add vit_base_patch16_224_pretrain

* add vit_base_patch16_384_finetune

* add vit_base_patch16_384_finetune

* add vit_b_p16_224_finetune_imagenet

* add vit_b_p16_224_pretrain_imagenet

* add vit_b_p16_384_finetune_imagenet

* add vit

* add vit

* add vit head

* vit unitest

* keep up with ClsHead

* test vit

* add flag to determiine whether to calculate acc during training

* Changes related to mmcv1.3.0

* change checkpoint saving interval to 10

* add label smooth

* default_runtime.py recovery

* docformatter

* docformatter

* delete 2 lines of comments

* delete configs/_base_/schedules/imagenet_bs4096.py

* add configs/_base_/schedules/imagenet_bs2048_AdamW.py

* rename imagenet_bs4096.py to imagenet_bs2048_AdamW.py

* add helpers.py

* test vit hybrid backbone

* fix HybridEmbed

* use to_2tuple instead
 2021-04-16 19:22:41 +08:00
LXXXXR 7d618e6606
[Fix] Fix version (#209) 
* fix version

* add projects in openmmlab

* minor fix

* empty

* add mmocr

* empty

* empty

* fix linting
 2021-04-16 19:07:17 +08:00
agim-a 3affc481c8
[Fix] check for CLASSES in checkpoint meta (#207) 
- check for CLASSES in checkpoint meta when key meta does not exists
 2021-04-15 22:19:23 +08:00
whcao 1cde6f6e65
[Feature] Add cutmix option (#198) 
* Add cutmix option

* fix code style

* add some annotations

* add annotation about custom_hooks

* check constraint of alpha > 0

* add test cutmix

* fix code style

* add cutmix to configs/models

* add cutmix to configs/resnet

* flake8

* empty
 2021-04-14 21:27:42 +08:00
mzr1996 b7b520881f
Update CONTRIBUTING.md according to mmcv (#210) 
* Update CONTRIBUTING.md according to mmcv

* Docstring formatting by docformatter

* Update openmmlab website.
 2021-04-14 21:22:37 +08:00
LXXXXR 3f085026cf
[Feature] Add Cifar100 config (#208) 
* add config for cifar100

* fix doc
 2021-04-13 20:15:29 +08:00
whcao af83e981ac
[Bug]Fix label smooth bug (#203) 
* add convert_to_one_hot

* add test_label_smooth_loss

* add my label_smooth_loss

* fix CELoss bug

* test new label smooth loss

* LabelSmoothLoss downward compatibility

* add some comments

* remove the old version of LabelSmoothLoss

* add some comments

* add some comments

* add some comments

* add label smooth to config
 2021-04-13 13:53:56 +08:00
whcao dcf61173f6
[Feature]Add cal_acc to cls_head.py (#206) 
* add cal_acc to cls_head.py

* test ClsHead with cal_acc

* 4 spaces indentation
 2021-04-13 13:52:14 +08:00
LXXXXR 5195932952
[Feature] Support random augmentation (#201) 
* support random augmentation

* minor fix on posterize

* minor fix on posterize

* minor fix on cutout

* minor fix on cutout

* fix bug in solarize add

* revised according to comments
 2021-04-09 14:02:50 +08:00
QingChuanWS 2bd28435cf
Add simplify option in pytorch2onnx.py and Update doc (#200) 
* add simplify option in pytorch2onnx.py and update relevant doc

* update doc about pytorch2onnx

* update code

* update docs
 2021-04-09 13:05:25 +08:00
LXXXXR 4d1fb1a662
[Feature] ColorJitter and Lighting (#190) 
* add configs

* remove config

* add color jitter and lighting

* revised according to comments
 2021-04-02 19:23:39 +08:00
QingChuanWS d259306013
Add pytorch to onnx and onnx to tensorrt doc (#180) 
* add pytorch2onnx and onnx2trt doc

* fix the typo in deploy doc

* fix bug of onnx2trt.md

* Remove redundant install steps in onnx2trt.md

* update doc based on code review

* update index.rst
 2021-04-01 15:56:55 +08:00
LXXXXR 1f6549eeee
bump version to 0.10.0 (#194) 2021-04-01 10:39:18 +08:00
LXXXXR 93cd960466
[Feature] Support AutoAug, AutoContrast, Equalize, Contrast, Brightness and Sharpness (#179) 
* add AutoContrast, Equalize, Contrast, Brightness and Sharpness pipelines

* add ImageNetPolicy

* add configs

* add unittest

* remove config

* rerun CI

* rerun CI

* [Fix] Update pip install mmcv command in ci (#187)

* update pip install mmcv command in ci

* update pip install mmcv command in ci

* fix ci

* fix ci
 2021-03-30 15:38:55 +08:00
LXXXXR d412841e08
[Fix] Update pip install mmcv command in ci (#187) 
* update pip install mmcv command in ci

* update pip install mmcv command in ci

* fix ci

* fix ci
 2021-03-29 19:57:27 +08:00
LXXXXR b507fc207b
[Fix] Use MMCV's EvalHook in MMClassification (#182) 
* use eval_hook in mmcv

* minor fix
 2021-03-25 17:38:51 +08:00
LXXXXR 77e88844cd
[Bug] Bug in vgg config (#181) 
* fix vgg_voc config

* fix vgg_voc config
 2021-03-25 15:50:01 +08:00
LXXXXR d3dceddf31
add model zoo for fp16 (#184) 2021-03-25 15:49:26 +08:00
ftbabi 49ece048e0
Add tutorials (#176) 
* Add tutorials using MobileNetV2

* Modify text and use ResNet50

Co-authored-by: wangruohui <12756472+wangruohui@users.noreply.github.com>
 2021-03-17 16:29:21 +08:00
LXXXXR e76c5a368d
[Feature] Support fp16 training (#178) 
* change mmcls fp16 to mmcv hook

* support fp16

* clean unnessary stuff
 2021-03-17 15:53:55 +08:00
QingChuanWS 8eb845e718
support dynamic shape export to onnx (#175) 
* support dynamic shape export to onnx

* update the code and reset the simple_test() of image.py

* fix dynamic_axes name is not a valid input/output name
 2021-03-17 15:50:03 +08:00
LXXXXR f3b9380cbc
[Feature] Support solarize and posterize pipelines (#172) 
* support Solarize and Posterize

* revised according to comments

* fix conflicts
 2021-03-09 19:45:13 +08:00
LXXXXR 9614787fc4
[Feature] Add color pipeline (#171) 
* add ColorTransform pipeline

* fix docstring

* minor change

* revised according to comments
 2021-03-09 19:28:50 +08:00
LXXXXR c8033ece8e
add invert pipeline (#168) 2021-03-02 16:46:57 +08:00
LXXXXR b1fa298a66
add Rotate pipeline (#167) 2021-03-01 21:49:07 +08:00
LXXXXR 7ca0ca271f
Bump version to 0.9.0 (#166) 
* update docs for version 0.9.0

* add missing #153
 2021-03-01 20:14:53 +08:00
LXXXXR bc50a7927b
reslove conflicts (#165) 2021-02-26 11:48:13 +08:00
LXXXXR 8c11c01fdb
[Feature] Add shear pipeline (#163) 
* half-done auto_augmentation

* remove auto_augcode and support shear pipeline

* fix typo

* fix typo

* use a non-square toy data instead
 2021-02-25 16:00:46 +08:00
David de la Iglesia Castro fb11a23cfe
Show results in test api (#162) 
* Initial support show in single_gpu_test

* Add result_show

* Add ori_filename meta_key

* test ori_filename

* Add show, show-dir, show-options
 2021-02-25 16:00:22 +08:00
LXXXXR 04cdbe4d66
remove unnecessary addopts (#164) 2021-02-25 14:09:03 +08:00
ftbabi bdd6b01ae7
[Feature] Add "mixup" from Bag of Tricks (#160) 
* Add mixup option

* Modify the structure of mixup and add configs

* Clean configs

* Add test for mixup and SoftCrossEntropyLoss

* Add simple test for ImageClassifier

* Fix bug in test_losses.py

* Add assertion in CrossEntropyLoss
 2021-02-25 14:06:58 +08:00
LXXXXR 2d7dc997e6
[Fix] Fix vgg configs and upload checkpoints (#161) 
* fix configs for vgg

* change lr

* rename configs for vgg

* fix vgg in model zoo
 2021-02-20 19:37:38 +08:00
RunningLeon f0d358e340
add onnx2tensorrt (#153) 2021-02-05 17:48:37 +08:00
LXXXXR ddc2a14177
[Feature] Add --eval-options in test.py (#158) 
* add --eval-options in test.py

* fix typo

* revise according to commnets
 2021-02-05 17:46:43 +08:00
LXXXXR 7f49632d7c
Bump version to 0.8.0 (#156) 
* update docs to bump version

* minor change

* revise acording to comments
 2021-01-31 17:50:40 +08:00
LXXXXR a7d4739d2b
[Fix] Fix test.py (#155) 
* modeify test.py

* modeify test.py

* revise according to comments
 2021-01-31 15:53:43 +08:00