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
fc82c31b2f
fix RandomCrop and RandomResizedCrop ( #151 )
2021-01-25 20:28:16 +08:00
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
13c1210741
[Feature] Add thrs in eval_metrics ( #146 )
...
* support thr
* replace thrs with thr
* fix docstring
* minor change
* revise according to comments
* revised according to comments
* revise according to comments
* rewrite basedataset.evaluate to avoid duplicate calculation
* minor change
* change thr to thrs
* add more unit test
2021-01-25 17:54:22 +08:00
87d4a1fcea
Update README.md ( #150 )
2021-01-25 17:16:57 +08:00
aa50103442
fix typo in configs ( #147 )
2021-01-21 14:47:02 +08:00
8e990b5654
[Feature] Support support and class-wise evaluation results ( #143 )
...
* support support, support class-wise evaluation results and move eval_metrics.py
* Fix docstring
* change average to be non-optional
* revise according to comments
* add more unittest
2021-01-19 16:42:16 +08:00
4f8fc9cbf3
[Feature] Add analyze_results.py to select imgs for success/fail ( #142 )
...
* add analyze_results.py to select imgs for success/fail
* minor change
* save imgs
2021-01-19 15:49:14 +08:00
b98b317012
fix bug in vgg weight_init ( #140 )
2021-01-15 10:23:11 +08:00
c8109d1eca
Update stat to classify papers ( #139 )
...
* update stat to classify papers
* modify tag of papers
2021-01-14 16:48:05 +08:00
63f38988eb
[Fix] Fix optional issues in docstring ( #138 )
...
* fix optional issue in docstring
* revised according to comments
* add optional
2021-01-14 11:09:08 +08:00
6916f33d56
[Feature] Add asymmetric loss for multilabel task ( #132 )
...
* add asymmetric loss
* minor change
* fix docstring
* do not apply sum over classes and fix docstring
* fix docstring
* fix weight shape
* fix weight shape
* add reference
* fix linkting issue
Co-authored-by: Y. Xiong <xiongyuxy@gmail.com>
2021-01-11 11:22:22 +08:00
d062cdef0a
[Docs] Add tags in README.md ( #137 )
...
* add tag in README.md
* fix linting issue
2021-01-11 11:10:06 +08:00
c07e60dd7d
[Feature] Support PASCAL VOC 2007 dataset for multilabel task ( #134 )
...
* support voc
* minor change
2021-01-11 11:08:59 +08:00
194ab7efda
[Feature] Add bce loss for multilabel task ( #130 )
...
* add bce loss for multilabel task
* minor change
* apply class wise sum
* fix docstring
* do not apply sum over classes and fix docstring
* fix docstring
* fix weight shape
* fix weight shape
* fix docstring
* fix linting issue
Co-authored-by: Y. Xiong <xiongyuxy@gmail.com>
2021-01-11 11:05:24 +08:00
agim-a
whcao
mzr1996
LXXXXR
QingChuanWS
ftbabi
QingChuanWS
David de la Iglesia Castro
RunningLeon
Cheng-Yen Yang
congee