mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
154 Commits
8 Branches
41 Tags
32 MiB
Commit Graph

154 Commits (bc1b08ba416a6417226b3de98c1e39e1ff594b61)
 

Author SHA1 Message Date
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
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
David de la Iglesia Castro 6b0b3da92a
Add note about ground-truth labels range (#44) 
* Add note about ground-truth labels range

* Fix
 2020-09-09 16:21:25 +08:00
anthracene fef5202257
Fix crash in CenterCrop transform when image is greyscale (#40) 2020-08-31 00:04:39 +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 91227e59cf
fix mobilenetv2 convert tools (#34) 
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
 2020-08-13 14:05:34 +08:00
Xiaojie Li 59abed3c39
Update pretrained model (#33) 
* update model_zoo

* update performance

* update shufflenet v2

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
 2020-08-13 14:04:57 +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
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
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
Lei Yang 3e2d3d726a
Merge pull request #21 from yl-1993/dev/resize_backend 
Add PIL backend for resize transform
 2020-08-07 21:50:51 +08:00
yl-1993 0009591aa3 add unittest for backends 2020-08-05 16:53:35 +08:00
yl-1993 a0c814fecd add doc for backend 2020-08-05 15:35:40 +08:00
Lei Yang 360ae80591
rename detectors to classifiers (#19) 2020-08-05 00:43:13 +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 34efbc8a9d add pil backend for resize 2020-08-03 12:52:35 +08:00
Lei Yang 557a364d25
replace urlretrieve with urlopen (#13) 2020-07-27 22:20:03 +08:00
Lei Yang 8e1ef6af26
rm bibtex and contact (#15) 2020-07-27 22:18:05 +08:00
LiJiezhi 8247704afc
Update README.md (#7) 2020-07-21 19:28:50 +08:00
Lei Yang 22632ae24c
correct module name from mmdet to mmcls (#8) 2020-07-21 19:28:14 +08:00
Kai Chen 0fb9b2f140
Update readme (#3) 
* add badges and license

* minor format
 2020-07-13 12:55:34 +08:00
Kai Chen 18c5dfc5f9
Merge pull request #2 from yl-1993/fix/docs 
Add requirements during readthedocs build
 2020-07-13 01:33:22 +08:00
yl-1993 f599e1ce3d add .readthedocs.yml 2020-07-13 01:08:40 +08:00
yl-1993 41fae9b66b add docs conf 2020-07-13 00:47:13 +08:00
yl-1993 670ef43d49 rm gitlab-ci 2020-07-12 23:33:52 +08:00
yl-1993 71e3b86ed3 add github action (manually squash) 2020-07-12 00:07:05 +08:00
yl-1993 4e6875d44e add index.rst for readthedocs 2020-07-12 00:06:57 +08:00
yanglei 5c88675a71 Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
yanglei 563fa63d06 Add model_zoo docs 2020-07-12 00:06:57 +08:00
yanglei 9a661ef981 Add ResNet_CIFAR 2020-07-12 00:06:56 +08:00
yanglei e29882c8af fix old register 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
yanglei 6968ad5b3b Add RandomFlip 2020-07-12 00:06:56 +08:00
yanglei 45812e87bd add weighted_loss 2020-07-12 00:06:55 +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 4bc2268f48 Add tutorial docs 2020-07-12 00:06:54 +08:00
yanglei 53f5568119 Add README 2020-07-12 00:06:54 +08:00
yanglei 8995e16834 Add AlexNet and LeNet5 2020-07-12 00:06:54 +08:00
yanglei 85399b397d Add installation docs 2020-07-12 00:06:53 +08:00
yanglei cf3a530c73 Add two useful tools from mmdet 2020-07-12 00:06:53 +08:00
yanglei d967b94439 Add getting_started docs 2020-07-12 00:06:53 +08:00
yanglei fbc68bf266 Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
yanglei 7255f9298e Merge branch 'dev/aug_resize_centercrop' into 'master' 
Add Resize and CenterCrop

See merge request open-mmlab/mmclassification!27
 2020-07-07 17:07:58 +08:00