46 Commits

Author SHA1 Message Date
Ma Zerun
9f7168e770
Add RegNetX configs (#269)
* Add RegNetX configs

* Fix typo in RegNet readme.

* Use comment to mark tag.
2021-06-04 14:29:13 +08:00
LXXXXR
c68bf9f2ba
fix resnetv1d configs and update ckpts (#288) 2021-06-03 16:30:29 +08:00
LXXXXR
df3d2d47f0
update resnext ckpts (#283) 2021-06-01 14:57:46 +08:00
LXXXXR
5ff158cec9
[Bug] Fix Cifar color channel (#280)
* fix bug

* updatedoc accordingly
2021-05-31 14:15:22 +08:00
Mingqiang Ning
9a66113d20
add attribute 'test' in dict 'data' (#264)
* add attribute 'test' in dict 'data'

* Update lenet5_mnist.py

 delete spaces around '='.

* init

* Delete custom directory
2021-05-31 10:19:27 +08:00
LXXXXR
98b6c4afa4
update cifar config (#275) 2021-05-27 15:20:25 +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
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
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
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
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
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
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
LXXXXR
07bb15e5fd
[Feature] Add heads and config for multilabel task (#145)
* resolve conflicts
add heads and config for multilabel tasks

* minor change

* remove evaluating mAP in head

* add baseline config

* add configs

* reserve only one config

* minor change

* fix minor bug

* minor change

* minor change

* add unittests and fix docstrings
2021-01-25 18:10:14 +08:00
Cheng-Yen Yang
87d4a1fcea
Update README.md (#150) 2021-01-25 17:16:57 +08:00
LXXXXR
aa50103442
fix typo in configs (#147) 2021-01-21 14:47:02 +08:00
congee
c8109d1eca
Update stat to classify papers (#139)
* update stat to classify papers

* modify tag of papers
2021-01-14 16:48:05 +08:00
LXXXXR
d062cdef0a
[Docs] Add tags in README.md (#137)
* add tag in README.md

* fix linting issue
2021-01-11 11:10:06 +08:00
LXXXXR
736eec1fd2
Fix links in docs (#118)
* fix links in docs

* Revised according to comments

* revised according to comments
2020-12-21 15:29:12 +08:00
LXXXXR
668292f5ab
Add README.md in configs (#117)
* rearrange config files

* rearrange config files

* add README.md in configs

* rename
2020-12-18 20:04:54 +08:00
LXXXXR
384ff95479
Rearrange config files (#116)
* rearrange config files

* rearrange config files

* add README.md in configs

* Revert "add README.md in configs"

This reverts commit 68ff996f9157843ae9d188c067db8ee86c4b387d.

* rename
2020-12-18 16:59:50 +08:00
WRH
44bbc71e14
Fix bug and optimize MNIST config (#98)
* add simple_test to ClsHead

* optimize lenet training config

* recover path setting
2020-11-26 15:27:04 +08:00
David de la Iglesia Castro
4f4f2957ef
Use build_runner (#54)
* Use build_runner in train api

* Support iter in eval_hook

* Add runner section

* Add test_eval_hook

* Pin mmcv version in install docs

* Replace max_iters with max_epochs

* Set by_epoch=True as default

* Remove trailing space

* Replace DeprecationWarning with UserWarning

* pre-commit

* Fix tests
2020-10-15 21:12:50 +08:00
David de la Iglesia Castro
b577594d3d fix outdated configs 2020-10-01 16:15:14 +02:00
Lei Yang
9547e7b7a5
Add model inference (#16)
* add model inference on single image

* rm --eval

* revise doc

* add inference tool and demo

* fix linting

* rename inference_image to inference_model

* infer pred_label and pred_score

* fix linting

* add docstr for inference

* add remove_keys

* add doc for inference

* dump results rather than outputs

* add class_names

* add related infer scripts

* add demo image and the first part of colab tutorial

* conduct evaluation in dataset

* return lst in simple_test

* compuate topk accuracy with numpy

* return outputs in test api

* merge inference and evaluation tool

* fix typo

* rm gt_labels in test conifg

* get gt_labels during evaluation

* sperate the ipython notebook to another PR

* return tensor for onnx_export

* detach var in simple_test

* rm inference script

* rm inference script

* construct data dict to replace LoadImage

* print first predicted result if args.out is None

* modify test_pipeline in inference

* refactor class_names of imagenet

* set class_to_idx as a property in base dataset

* output pred_class during inference

* remove unused docstr
2020-09-30 19:00:20 +08:00
Lei Yang
bc1b08ba41
Add VGG and pretained models (#27)
* add vgg

* add vgg model coversion tool

* fix out_indices and docstr

* add vgg models in configs

* add params, flops and accuracy in docs

* add pretrained models url

* use ConvModule and refine var names

* update vgg conversion tool

* modify bn config

* add docs for arch_setting

* add unit test for vgg

* rm debug code

* update vgg pretrained models
2020-09-29 17:49:42 +08:00
Jerry Jiarui XU
8d3acce307
Add ResNeSt (#25)
* Add ResNeSt

* fixed test

* refactor

* add ResNeSt base

* update modelzoo

* update modelzoo

* Add S-200,S269 _base_
2020-09-22 10:41:51 +08:00
Lei Yang
4b46fd6dc7
Add CIFAR10 configs and models (#38)
* rename cifar configs

* add cifar10 configs

* fix linting

* add params, flops and accuracy in docs

* add cifar10 pretrained models and logs

* use oss-accelerate url

* del unused cifar10 configs
2020-08-26 16:34:15 +08:00
Lei Yang
b2fc837e84
Rename configs with samples_per_batch x gpus (#37)
* rename configs with samples_per_batch x gpus

* revise docs accordingly
2020-08-23 23:53:51 +08:00
Xiaojie Li
18ac0f0fa3
Update shufflenet config (#31)
* update shufflenet config

* remove old config

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 10:46:39 +08:00
Xiaojie Li
414ede4ca4
fix coslr (#32)
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 10:45:41 +08:00
Xiaojie Li
608cea2469
fix (#30)
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 09:50:14 +08:00
yl-1993
7d1edd03c6 revise a part of configs to pil resize 2020-08-03 12:54:58 +08:00
yl-1993
8af1537569 add pil resize configs 2020-08-03 12:54:08 +08:00
yl-1993
71e3b86ed3 add github action (manually squash) 2020-07-12 00:07:05 +08:00
yanglei
5c88675a71 Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
chenkai
a99e6eb10d init commit 2020-05-21 21:21:43 +08:00