761c95925d
support mmcli ( #232 )
2021-05-01 22:24:16 +08:00
f970a36898
update pytorch2onnx.md table ( #229 )
2021-04-30 17:18:06 +08:00
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
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
5fb9ced5f9
[Docs] Change tag into comment ( #226 )
...
* change tags
* remove modelzoo_statistics
2021-04-28 17:08:57 +08:00
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
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
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
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
d72d233f9b
fix resnext resize ( #222 )
2021-04-25 16:26:34 +08:00
759aa92437
[Bug]add cutmix_prob to config ( #220 )
...
* add cutmix_prob to config
* empty
2021-04-21 21:57:09 +08:00
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
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
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
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
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
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
3f085026cf
[Feature] Add Cifar100 config ( #208 )
...
* add config for cifar100
* fix doc
2021-04-13 20:15:29 +08:00
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
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
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
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
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
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
1f6549eeee
bump version to 0.10.0 ( #194 )
2021-04-01 10:39:18 +08:00
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
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
b507fc207b
[Fix] Use MMCV's EvalHook in MMClassification ( #182 )
...
* use eval_hook in mmcv
* minor fix
2021-03-25 17:38:51 +08:00
77e88844cd
[Bug] Bug in vgg config ( #181 )
...
* fix vgg_voc config
* fix vgg_voc config
2021-03-25 15:50:01 +08:00
d3dceddf31
add model zoo for fp16 ( #184 )
2021-03-25 15:49:26 +08:00
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
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
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
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
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
c8033ece8e
add invert pipeline ( #168 )
2021-03-02 16:46:57 +08:00
b1fa298a66
add Rotate pipeline ( #167 )
2021-03-01 21:49:07 +08:00
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
bc50a7927b
reslove conflicts ( #165 )
2021-02-26 11:48:13 +08:00
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
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
04cdbe4d66
remove unnecessary addopts ( #164 )
2021-02-25 14:09:03 +08:00
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
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
f0d358e340
add onnx2tensorrt ( #153 )
2021-02-05 17:48:37 +08:00
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
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
a7d4739d2b
[Fix] Fix test.py ( #155 )
...
* modeify test.py
* modeify test.py
* revise according to comments
2021-01-31 15:53:43 +08:00
a225cb6bdb
fix img_metas bug ( #152 )
2021-01-26 11:24:08 +08:00
3e5a9513be
[Bug] Fix bug in Collect ( #149 )
...
* fix bug in Collect
* add metakeys
2021-01-25 20:29:28 +08:00
LXXXXR
QingChuanWS
whcao
mzr1996
WRH
agim-a
ftbabi
QingChuanWS
David de la Iglesia Castro
RunningLeon