Commit Graph

77 Commits (a225cb6bdbf027250b75e725e2536c02f48ce27c)

Author SHA1 Message Date
LXXXXR b98b317012
fix bug in vgg weight_init (#140) 2021-01-15 10:23:11 +08:00
LXXXXR 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
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
Xiaojie Li 7af9419ffa
Fix init_weights in 'shufflenet_v2.py'. (#29)
* fix init_weights in shufflenetv2

* fix doc

* fix doc

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 09:49:41 +08:00
Lei Yang 408d92bcbe
revise docstr of alexnet to correct input_size (#26) 2020-08-12 16:14:19 +08:00
yanglei 9a661ef981 Add ResNet_CIFAR 2020-07-12 00:06:56 +08:00
lixiaojie c1d0090700 Change the init_weight for shufflenet models 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 8995e16834 Add AlexNet and LeNet5 2020-07-12 00:06:54 +08:00
yanglei 17092d8be4 Fix the mid_channels of SEResNeXt 2020-07-07 11:25:48 +08:00
lixiaojie eb24a94b68 add mobilenetv2 convert code 2020-07-05 21:59:22 +08:00
lixiaojie 046aa5c003 add shufflenet_v2 convert code 2020-07-03 19:21:18 +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
lixiaojie 50208498dd Update mobilenetv2 2020-06-22 19:40:31 +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
chenkai c168aa786e Merge branch 'fix/resnet_zero_init_residual' into 'master'
Set default zero_init_residual of ResNet to True

See merge request open-mmlab/mmclassification!22
2020-06-16 22:09:48 +08:00
louzan deee5d61d7 add mobilenetv2 2020-06-16 14:37:03 +08:00
yanglei 800ace3137 set default zero_init_residual to True 2020-06-16 11:36:54 +08:00
lixiaojie fcb1ad80e4 merge master 2020-06-16 00:49:55 +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 3bf971238a fix grammar 2020-06-15 16:29:36 +08:00
lixiaojie 402bedf489 update assertion message 2020-06-15 16:26:24 +08:00
lixiaojie c3576c77fe fix docstring 2020-06-15 16:21:58 +08:00
lixiaojie b0a0248404 fix grammar & indent problem 2020-06-15 16:16:27 +08:00
lixiaojie 3bf34be1ed fix indent problem 2020-06-15 16:11:51 +08:00
lixiaojie 62f65153f9 update shufflenet_v2.py 2020-06-15 15:15:19 +08:00
lixiaojie 24ef3b2f61 fix linting 2020-06-15 02:34:22 +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 a5b53ada94 use standard docstr for basic and bottleneck block 2020-06-14 13:29:47 +08:00
chenkai 335047083e Merge branch 'align_resnet_accuracy' into 'master'
Align resnet accuracy

See merge request open-mmlab/mmclassification!16
2020-06-14 12:11:07 +08:00
yanglei 76f91510b3 Add ResNeXt 2020-06-14 12:08:37 +08:00
yanglei 7ec1429e1b set default value of norm_eval and zero_init_residual to False 2020-06-14 11:56:18 +08:00
lixiaojie 461e07d65d Update default value 2020-06-14 03:51:42 +08:00
lixiaojie fb3934fd2c fix norm_cfg&rewrite 2020-06-14 01:23:40 +08:00
lixiaojie 02e1c246b4 Update shufflenet_v2.py 2020-06-14 01:13:10 +08:00
lixiaojie e3e980d84e rewrite 2020-06-14 00:46:22 +08:00