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

279 Commits (a06827bc0872fee86144ece4394ae72ea435de95)
 

Author SHA1 Message Date
Ma Zerun a06827bc08
Build chinese translation in readthedocs (#267) 
* change document layout

* Add Chinese docs build file

* Add switch language option
 2021-05-29 10:00:37 +08:00
LXXXXR 98b6c4afa4
update cifar config (#275) 2021-05-27 15:20:25 +08:00
Yinhao Li b67a11c548
Fix kwargss to kwargs. (#274) 2021-05-26 19:27:30 +08:00
LXXXXR 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
Y. Xiong 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
Ma Zerun 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
LXXXXR ace09510f7
[Bug]Fix typo (#255) 
* fix typo

* rerun ci
 2021-05-17 10:42:31 +08:00
LXXXXR 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
Wenwei Zhang 5ee08767f2
inherits mmcv registry (#252) 2021-05-14 23:36:56 +08:00
LXXXXR 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.

* revise according to comments
 2021-05-12 15:21:55 +08:00
LXXXXR f415c49bc3
fix MANIFEST.in (#250) 2021-05-12 15:04:58 +08:00
mzr1996 9e93feabac
[Docs] Fix error in new_dataset.md and add Chinese translation of finture.md, new_dataset.md (#243) 
* Fix error in new_dataset.md

* Add Chinese Translation of finture.md, new_dataset.md
 2021-05-10 17:17:37 +08:00
mzr1996 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
Y. Xiong 22b40696d6
add tempdir check (#242) 2021-05-10 17:12:43 +08:00
Forsaken 0a92a81198
Update analyze_results.py (#237) 
correct spelling mistake(sucess->success)
 2021-05-10 15:04:08 +08:00
RangiLyu 6a50a2864b
fix typo in unit test (#238) 2021-05-10 15:03:39 +08:00
mzr1996 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
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