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

160 Commits (8ef5aeda408d90bdcd94fa412d81a4088289cc0b)

Author SHA1 Message Date
LXXXXR 21fd5019fb
add macro-averaged precision,recall,f1 options in evaluation (#93) 
* add macro-averaged precision,recall,f1 options in evaluation

* remove unnecessary comments

* Revise according to comments

* Revise according to comments
 2020-11-25 16:13:54 +08:00
LXXXXR 7636409b3b
add get_class in base_dataset (#85) 
* modify base_dataset

* revise according to the comments
 2020-11-12 14:22:02 +08:00
David de la Iglesia Castro df2189c6f6
Add missing CLASSES argument to dataset wrappers (#66) 
* Add missing classes in dataset wrappers

* Update tests
 2020-10-15 21:25:53 +08:00
David de la Iglesia Castro 4f4f2957ef
Use build_runner (#54) 
* Use build_runner in train api

* Support iter in eval_hook

* Add runner section

* Add test_eval_hook

* Pin mmcv version in install docs

* Replace max_iters with max_epochs

* Set by_epoch=True as default

* Remove trailing space

* Replace DeprecationWarning with UserWarning

* pre-commit

* Fix tests
 2020-10-15 21:12:50 +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
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
lixiaojie 0def3a56b6 fix test 2020-06-05 15:52:22 +08:00
lixiaojie fa49911440 reformat 2020-06-04 02:42:53 +08:00
lixiaojie 0667136f05 update test 2020-06-04 02:22:53 +08:00
lixiaojie 83cc9fc279 fix flake8 2020-06-03 15:54:37 +08:00
lixiaojie 2aaff0e4f2 fix & add test 2020-06-03 15:51:17 +08:00