c2f01e0dcd
[Feature] Add torchscript deployment ( #279 )
...
* add torchscript deploy
* fix lint
* add check and delete \
2021-06-12 21:50:48 +08:00
dbddde52ef
[Feature] TensorRT test tools. ( #284 )
...
* first commit
* update resnext result
* update docs
* update docstring
2021-06-12 21:47:10 +08:00
5e1a02103f
[Feature]Delete comments ( #298 )
...
* add mytrain.py for test
* test before layers
* test attr in layers
* test classifier
* delete mytrain.py
* delete comments
2021-06-12 21:45:22 +08:00
84a939f858
Refactor LabelSmoothLoss ( #285 )
...
* Refector label smooth loss, now support mode `original`, `classy_vision`
and `multi_label`.
* Add unittests for label smooth loss.
* Improve docstring of LSR
2021-06-12 21:32:18 +08:00
6de635a81c
[Bug] Missing test data when num_imgs can not be evenly divided by num_gpus ( #299 )
...
* fix bug in test
* remove unneccesary code
2021-06-11 15:53:23 +08:00
4ca21c7d03
[WIP] Refactoring weights initialization ( #270 )
...
* [WIP] Refactoring weights initialization
* fix lint and constant init cfg
* fix pretrained bug
* fix typo
* fix isort
* revise model utils
2021-06-10 10:54:34 +08:00
27a49a9646
bump version to v0.12.0 ( #287 )
2021-06-03 11:42:34 +08:00
2c9e12f850
[Feature] Add an argument `efficientnet_style` to `RandomResizedCrop` and `CenterCrop` ( #268 )
...
* add config for resnest test
* fix config
* add label smoothing
* add memcached
* minor fix
* fix bug
* fix config
* add config
* minor fix
* fix configs
* use EResize
* change interpolation
* add more configs
* add docsting
* add unittest
* remove unnecessary changes
* minor fix
* add more docstring
* fix linting
* refactor
* add resize in crop to ensure crop size is output size
* fix bug and add comments
* fix bug
2021-05-31 14:10:57 +08:00
b23b319f56
fix version ( #276 )
2021-05-29 10:47:16 +08:00
bd9411d743
[Bug] Download dataset only on rank 0 ( #273 )
...
* only download dataset on rank 0
* download only on rank 0
* fix bug
* fix error message
2021-05-29 10:45:58 +08:00
b67a11c548
Fix kwargss to kwargs. ( #274 )
2021-05-26 19:27:30 +08:00
dac090162d
Bump version to v0.11.1 ( #256 )
...
* bump version to v0.11.1
* minor fix
* minor fix
* minor fix
* minor fix
2021-05-21 16:36:08 +08:00
82e3937174
[Fix] Only allow directory operation when rank==0 when testing ( #258 )
...
* only allow dir operation when rank==0
* move check dir to multi_gpu_test
2021-05-21 14:04:46 +08:00
09597e5a4c
Add transform `RandomErasing` ( #248 )
...
* Add transform `RandomErasing`.
* Add unittests of `RandomErasing`
* Fix typo in docstring
* Improve docstring and unittests.
2021-05-19 22:35:26 +08:00
dc296f64c6
[Fix] Fix multi-node test tmp dir ( #251 )
...
* fix multi-node test tmp dir
* fix mmcv version
2021-05-16 21:53:13 +08:00
5ee08767f2
inherits mmcv registry ( #252 )
2021-05-14 23:36:56 +08:00
8c90a879ce
[Fix] Fix magnitude_range in RandAug ( #249 )
...
* add increasing in solarize and posterize
* fix linting
* Revert "add increasing in solarize and posterize"
This reverts commit 128af36e9b
2021-05-12 15:21:55 +08:00
a3b8d6015d
[Feature] Add RandAUG magnitude noise ( #240 )
...
* Add paramater magnitude_std in RandAugment to allow randomly movement of magnitude_value
* Add unittest for magnitude_std
* Improve docstring of magnitude_std
2021-05-10 17:13:41 +08:00
22b40696d6
add tempdir check ( #242 )
2021-05-10 17:12:43 +08:00
8128900a12
GlabelAveragePooling support 1d, 2d and 3d by param, and add neck test ( #236 )
...
* GlabelAveragePooling support 1d, 2d and 3d by param, and add neck test
* Imporve neck test
* Change 'mode' attribute in GAP to 'dim', and add docstring
2021-05-10 15:00:50 +08:00
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
37167158e7
bump version to v0.11.0 ( #233 )
2021-05-01 22:26:39 +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
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
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
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
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
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
b507fc207b
[Fix] Use MMCV's EvalHook in MMClassification ( #182 )
...
* use eval_hook in mmcv
* minor fix
2021-03-25 17:38:51 +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
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
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
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
AllentDan
q.yao
whcao
Ma Zerun
LXXXXR
Miao Zheng
Yinhao Li
Y. Xiong
Wenwei Zhang
mzr1996
QingChuanWS
WRH
agim-a
David de la Iglesia Castro
ftbabi