mirrors/mmclassification
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 21:53:55 +08:00
Code Issues Projects Releases Wiki Activity
612 Commits 8 Branches 41 Tags
Commit Graph

213 Commits

Author SHA1 Message Date
Ezra-Yu
e9342d9e4c Add multi_label heads 2022-07-18 11:11:13 +08:00
yingfhu
4fcd7ee072 Add EfficientNetRandomCrop and EfficientNetCenterCrop 2022-07-18 11:11:13 +08:00
mzr1996
12c982f939 [Refactor] Refactor RepeatAugSampler. 2022-07-18 11:11:13 +08:00
yingfhu
e8d69cf2ff [Fix] adjust default dataset path to empty string 2022-07-18 11:11:13 +08:00
Ezra-Yu
7bca2516f0 Add CUB dataset 2022-07-18 11:11:13 +08:00
Ezra-Yu
daa6167336 Add Lighting, ColorJitter and Albumentations 2022-07-18 11:11:13 +08:00
Ezra-Yu
a9057e88c4 Add multi label metrics 2022-07-18 11:11:13 +08:00
yingfhu
62b046521e [Refactor] refactor ViTHead, DeiTHead, ConformerHead, StackedHead 2022-07-18 11:11:13 +08:00
Ezra-Yu
a82de04b67 Enhance browse datset 2022-07-18 11:11:13 +08:00
mzr1996
375fe68f12 Imporve according to comments 2022-07-18 11:11:13 +08:00
mzr1996
f0cab33e09 Force label in ClsDatasample long type. 2022-07-18 11:11:13 +08:00
mzr1996
f3299b4ca2 [Refactor] Refactor batch augmentations 2022-07-18 11:11:13 +08:00
mzr1996
dd660ed99e [Improve] Clean useless code and reduce unit tests memory usage. 2022-07-18 11:11:13 +08:00
yingfhu
125b74d4ca [Refactor] Refactor MNIST and FashionMNIST dataset. 2022-07-18 11:11:13 +08:00
mzr1996
ee6e585e41 Mazerun/base classifier 2022-07-18 11:11:13 +08:00
Ezra-Yu
d6fa480915 [Refactor] Add VOC dataset and mutil_label dataset 2022-07-18 11:11:13 +08:00
mzr1996
522ab1fd84 [Feature] Add Visualization hook 2022-07-18 11:11:04 +08:00
yingfhu
c78b5597d8 [Refactor] refactor randomCrop, randomResizeCrop and CenterCrop。 2022-07-18 10:57:17 +08:00
mzr1996
c0feadf546 Refactor RandomErasing 2022-07-18 10:53:56 +08:00
mzr1996
4f28b9dd63 [Refactor] refactor AutoAugment and RandAugment 2022-07-18 10:53:56 +08:00
mzr1996
3b6d44ea05 [Refactor] Refactor CIFAR10 and CIFAR100 dataset. 2022-07-18 10:53:56 +08:00
mzr1996
995b1d0d58 [Refactor] Add ResizeEdge and refactor all dataset configs. 2022-07-18 10:53:56 +08:00
Ezra-Yu
4d4b22377d Add browse dataset 2022-07-18 10:53:56 +08:00
mzr1996
efaa93aae1 [Refactor] Refactor entry scripts train.py and test.py. 2022-07-18 10:53:56 +08:00
mzr1996
6ad75f0076 [Feature] Add metrics for single-label classification. 2022-07-18 10:53:28 +08:00
Ezra-Yu
93a27c8324 [Feature] Add PackClsInputs and use LoadImageFromFile, Resize & RandomFlip in MMCV. 2022-07-18 10:53:28 +08:00
mzr1996
0537c4d70c [Feature] Add ClsVisualizer. 2022-07-18 10:53:27 +08:00
mzr1996
27e685fe10 [Feature] Add BaseDataset, CustomDataset, ImageNet and ImageNet21k 2022-07-18 10:53:27 +08:00
mzr1996
98377df512 Update ClsDataSample design 2022-07-18 10:53:27 +08:00
mzr1996
ce2b40133b Add property, setter and deleter. 2022-07-18 10:53:27 +08:00
mzr1996
4bf54066c9 Add ClsDataSample 2022-07-18 10:53:27 +08:00
yingfhu
de002e455f [Refactor] Use MMLogger from MMEngine in get_logger and print_log. 2022-07-18 10:53:27 +08:00
Ezra-Yu
088d5b5add [Refactor] Inherit all registries from MMEngine and use inherited registries. 2022-07-18 10:53:27 +08:00
mzr1996
b5193a9029 [Fix] Reduce unit test memory usage of T2T-ViT 2022-05-16 17:01:30 +08:00
Hubert
59292b315a
[Feature] Support DenseNet. (#750) 
* init add densenet implementation

* Add config and converted models

* update meta

* add test for memory efficient

* Add docs

* add doc for jit

* Update checkpoint path

* Update readthedocs

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-04-28 13:06:03 +08:00
takuoko
df6edd7f5a
[Feature] Support VAN. (#739) 
* add van

* fix config

* add metafile

* add test

* model convert script

* fix review

* fix lint

* fix the configs and improve docs

* rm debug lines

* add VAN into api

Co-authored-by: Yu Zhaohui <1105212286@qq.com>
 2022-04-28 09:35:17 +08:00
Ma Zerun
504e71c3e0
[Enhance] Reduce the memory usage of unit tests for Swin-Transformer. (#759) 2022-04-22 17:41:15 +08:00
Weihao Yu
2eb27d0d4e
[Feature] Add PoolFormer backbone and checkpoints. (#746) 
* add PoolFormer

* fix some typos in PoolFormer

* fix lint error

* modify out_indices and gap

* fix typo

* fix lint

* fix typo

* fix typo in poolforemr README

* fix lint

* Update some paths

* Refactor freeze_stages method

* Add unit tests

* Fix lint

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-04-15 11:35:15 +08:00
Ezra-Yu
922a31a68e
[Feature] ADD RepMLP Backbone and checkpoints. (#709) 
* add repmlp

* refactor backbone code

* add checkpoint, readme, unit tests

* unchange demo

* improve docstring.

* update pic

* refactor patch embed

* refactor unit tests

* fix lint

* update tools

* update tools

* Update checkpoint path

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-03-30 20:40:25 +08:00
Ezra-Yu
292127098f
[Feature] Add CSPNet and backbone and checkpoints (#735) 
* add cspnet backbone

* try

* add backbones

* add cspnet

* add acc

* add unit tests

* chnage mode

* Update checkpoint path

* Fix typo

* Imporve CSPNet

* Update checkpoints

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-03-30 15:29:35 +08:00
Ma Zerun
d0d6f73fab
[Feature] Add CustomDataset. (#738) 
* Add custom dataset and refactor ImageNet dataset

* Add default CLASSES for CIFAR dataset

* Add unit tests

* Imporve according to comments
 2022-03-30 15:28:26 +08:00
huyu
a19c28fe95
[Enhance] Add get_cat_ids and get_gt_labels to KFoldDataset. (#721) 2022-03-23 15:57:36 +08:00
Hubert
04cb42a768
[Feature] Support ConvMixer. (#716) 
* basic support for ConvMixer

* simplify

* add data pipeine config for timm

* Add model readme and metafile

* add unittest for convmixer

* add copyright

* modify

* add tests

* update model

* add conv2dAdaptivePadding replacement

* update model index

* fix comments

* Update checkpoint path

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-03-23 15:23:57 +08:00
takuoko
aa522f4309
[Feature] Support CUB dataset. (#703) 
* support cub dataset

* support cub dataset

* fix train lint error

* add docs

* fix class label

Co-authored-by: Ezra-Yu <1105212286@qq.com>

* del debug code

* skip docformatter problem

* add unit tests

* add CUB baseline configs and chpts

* fix some typos

* fix name style

* update flops

Co-authored-by: Ezra-Yu <1105212286@qq.com>
 2022-03-16 16:22:28 +08:00
takuoko
c1534f9126
[Feature] Support resizemix. (#676) 
* add resizemix

* skip torch.__version__ < 1.7.0

* Update mmcls/models/utils/augment/resizemix.py

Co-authored-by: Ma Zerun <mzr1996@163.com>

* Update mmcls/models/utils/augment/resizemix.py

Co-authored-by: Ma Zerun <mzr1996@163.com>

* resize -> F.interpolate

* fix docs

* fix test

* add Copyright

* add argument interpolation

Co-authored-by: Ma Zerun <mzr1996@163.com>
 2022-03-07 12:11:20 +08:00
Ma Zerun
c708770b42
[Enhance] Support dynamic input shape for ViT-based algorithms. (#706) 
* Move `resize_pos_embed` to `mmcls.models.utils`

* Refactor Vision Transformer

* Refactor DeiT

* Refactor MLP-Mixer

* Refactor Swin-Transformer

* Remove `indexing` arg

* Support dynamic inputs for t2t_vit

* Add copyright

* Fix bugs in swin transformer

* Add `pad_small_maps` option

* Update swin transformer

* Handle `attn_mask` in checkpoints of swin

* Imporve by comments
 2022-03-03 13:10:12 +08:00
HumberMe
1214df083d
[Enhance] Use PyTorch official one_hot to implement convert_to_one_hot. (#696) 
* some change to mmcls/models/losses/utils.py:convert_to_one_hot()

* fixed problem: line too long

* fixed wrong output shape

* fixed lint PEP8 E128

* fix lint

* fix lint

* add unit tests

Co-authored-by: Ezra-Yu <1105212286@qq.com>
 2022-02-28 10:20:31 +08:00
Ma Zerun
5f7322c211
[Fix] Fix Conformer forward with irregular input size. (#686) 
* Auto pad the feature map to be divisible by 4.

* Add unit tests

* Use `trans_down_stride` instead of 4.
 2022-02-28 10:17:52 +08:00
Ma Zerun
28e3bc8fd9
[CI] Add Windows CI. (#708) 
* Update workflow to add Windows CI

* Add some optional requirements to requirements/optional.txt

* Update setup.py

* Update PyTorch version in Windows CI

* Update CI and requirement

* Replace `tempfile.NamedTemporaryFile` to avoid unit test error on
Windows

* Update tests

* Update CI

* Add OpenCV installation

* Update CI
 2022-02-28 10:06:08 +08:00
Ezra-Yu
7fcaedcbfb
[Feature] Add ResNetV1c. (#692) 
* add ResNetV1c

* add unit tests

* fix lint

* update docstring

* fix lint
 2022-02-23 11:36:33 +08:00