205 Commits

Author SHA1 Message Date
David de la Iglesia Castro
99115fddbc
Add albumentations (#45)
* Add Albu transform

* pre-commit

* Create optional.txt

* Update requirements.txt

* Update transforms.py
2020-09-22 17:35:39 +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
c905decddf
Resize according to short edge (#22)
* resize according to short edge

* revise doc for resize

* fix unitest for resize

* resize short_edge when second value of size is -1

* rename short_edge to short_side
2020-08-11 16:04:45 +08:00
yl-1993
0009591aa3 add unittest for backends 2020-08-05 16:53:35 +08:00
yanglei
9a661ef981 Add ResNet_CIFAR 2020-07-12 00:06:56 +08:00
yanglei
6968ad5b3b Add RandomFlip 2020-07-12 00:06:56 +08:00
lixiaojie
1d5f3d8c24 update shufflenet_v1 2020-07-12 00:06:55 +08:00
lixiaojie
f6260b33bf Add RandomCrop, RandomResizedCrop, RandomGrayscale, impad 2020-07-12 00:06:55 +08:00
yanglei
910d13330a Add Resize and CenterCrop 2020-07-07 17:07:58 +08:00
yanglei
17092d8be4 Fix the mid_channels of SEResNeXt 2020-07-07 11:25:48 +08:00
yanglei
f9b1f18f28 fix test_dataset 2020-07-06 00:43:22 +08:00
lixiaojie
046aa5c003 add shufflenet_v2 convert code 2020-07-03 19:21:18 +08:00
yanglei
ba789da5c8 Add commonly used datasets 2020-07-01 16:09:06 +08:00
chenkai
91a43c5bac Add RegNet 2020-07-01 14:21:45 +08:00
louzan
03b75789c6 Dev mobilenetv3 2020-06-30 15:50:36 +08:00
chenkai
02e11cc1f3 Refactoring for ResNet family 2020-06-25 11:57:50 +08:00
yangmingmin
29e17ab326 Dev se resnext 2020-06-19 11:45:42 +08:00
yangmingmin
f729a60f87 Dev se resnet 2020-06-17 14:20:20 +08:00
louzan
deee5d61d7 add mobilenetv2 2020-06-16 14:37:03 +08:00
lixiaojie
fcb1ad80e4 merge master 2020-06-16 00:49:55 +08:00
lixiaojie
425be621e0 fix unit test 2020-06-16 00:09:43 +08:00
lixiaojie
1112be1628 check assertion error 2020-06-16 00:07:04 +08:00
lixiaojie
66e9bb017d fix unit test 2020-06-16 00:03:45 +08:00
lixiaojie
8060e0f620 merge master 2020-06-15 20:54:52 +08:00
lixiaojie
1ce116a898 Merge branch 'master' of gitlab.sz.sensetime.com:open-mmlab/mmclassification into dev_shufflenetv1 2020-06-15 20:52:27 +08:00
lixiaojie
edceab13e3 modify self.layers 2020-06-15 20:52:17 +08:00
qianchen
5662daeea8 Merge branch 'dev_shufflenetv1' into 'master'
add shufflenetv1

See merge request open-mmlab/mmclassification!10
2020-06-15 20:49:44 +08:00
lixiaojie
8249ad7205 merge master 2020-06-15 20:46:30 +08:00
lixiaojie
cf85db0f7c Merge branch 'master' of gitlab.sz.sensetime.com:open-mmlab/mmclassification into dev_shufflenetv2 2020-06-15 20:42:22 +08:00
lixiaojie
703714b78e modify format of self.layers 2020-06-15 20:42:04 +08:00
wangshiguang
a1da2013ad add pat ci image 2020-06-15 17:43:40 +08:00
lixiaojie
29975930f9 Dev/backbone utils 2020-06-15 16:42:15 +08:00
lixiaojie
bb99ca5c66 fix grammar 2020-06-15 16:30:37 +08:00
lixiaojie
3bf971238a fix grammar 2020-06-15 16:29:36 +08:00
lixiaojie
62f65153f9 update shufflenet_v2.py 2020-06-15 15:15:19 +08:00
lixiaojie
58faa2381c fix norm_eval & merge master 2020-06-15 02:22:50 +08:00
lixiaojie
326df68bb3 fix norm_eval & merge master 2020-06-15 02:16:13 +08:00
yanglei
76f91510b3 Add ResNeXt 2020-06-14 12:08:37 +08:00
lixiaojie
85844a3a9e fix linting 2020-06-14 01:25:08 +08:00
lixiaojie
fb3934fd2c fix norm_cfg&rewrite 2020-06-14 01:23:40 +08:00
lixiaojie
e3e980d84e rewrite 2020-06-14 00:46:22 +08:00
lixiaojie
996f0a97e1 fix format 2020-06-12 18:07:53 +08:00
lixiaojie
eebb7561bf merge mastewr 2020-06-12 16:49:28 +08:00
lixiaojie
75858a3d3e merge master 2020-06-12 16:16:28 +08:00
yanglei
3a65f4529e Add Base and ResNet backbone 2020-06-12 14:40:27 +08:00
lixiaojie
4d5c39d1ee fix linting 2020-06-11 16:40:26 +08:00
lixiaojie
72bbfbc489 use ConvModule/build_*_layer from mmcv & use *_cfg & add _freeze_stages 2020-06-11 15:55:17 +08:00
lixiaojie
05e5173b6b complete the docstring & import from mmcv & range check for frozen_stages and out_indices 2020-06-10 15:43:38 +08:00
lixiaojie
16b4efbe06 add unit test 2020-06-07 23:33:47 +08:00
lixiaojie
98f5b49ffe add unit test 2020-06-07 22:40:36 +08:00