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

101 Commits (fe45b241c559a2689e836abbece715baa2912cfe)

Author SHA1 Message Date
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
Lei Yang 909a6b9c3f
Add get_cat_ids in BaseDataset (#72) 
* visualize results on image demo

* add get_cat_ids in BaseDataset
 2020-10-26 14:04:10 +08:00
Lei Yang 784987fe9f
Fix slurm evaluation error during training (#69) 
* visualize results on image demo

* set round_up to True during test
 2020-10-20 15:04:39 +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
Y. Xiong d369645e23
bump version to 0.6.0 (#62) 
* fix lint

* fix version path type

* del short_version

* setup: del ops; add python3.8

* del useless requirements

* setup: remove extra_requirements

* setup: totally remove build_requests

* setup: change to valid email
 2020-10-11 00:12:04 +08:00
Lei Yang 24fd4fb627
Visualize results on image demo (#58) 
* visualize results on image demo

* add matplotlib in requirements
 2020-10-10 16:33:27 +08:00
Lei Yang 9547e7b7a5
Add model inference (#16) 
* add model inference on single image

* rm --eval

* revise doc

* add inference tool and demo

* fix linting

* rename inference_image to inference_model

* infer pred_label and pred_score

* fix linting

* add docstr for inference

* add remove_keys

* add doc for inference

* dump results rather than outputs

* add class_names

* add related infer scripts

* add demo image and the first part of colab tutorial

* conduct evaluation in dataset

* return lst in simple_test

* compuate topk accuracy with numpy

* return outputs in test api

* merge inference and evaluation tool

* fix typo

* rm gt_labels in test conifg

* get gt_labels during evaluation

* sperate the ipython notebook to another PR

* return tensor for onnx_export

* detach var in simple_test

* rm inference script

* rm inference script

* construct data dict to replace LoadImage

* print first predicted result if args.out is None

* modify test_pipeline in inference

* refactor class_names of imagenet

* set class_to_idx as a property in base dataset

* output pred_class during inference

* remove unused docstr
 2020-09-30 19:00:20 +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
anthracene fef5202257
Fix crash in CenterCrop transform when image is greyscale (#40) 2020-08-31 00:04:39 +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
yl-1993 a0c814fecd add doc for backend 2020-08-05 15:35:40 +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
Kai Chen 0fb9b2f140
Update readme (#3) 
* add badges and license

* minor format
 2020-07-13 12:55:34 +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 8995e16834 Add AlexNet and LeNet5 2020-07-12 00:06:54 +08:00
yanglei fbc68bf266 Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +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
lixiaojie eb24a94b68 add mobilenetv2 convert code 2020-07-05 21:59:22 +08:00
yanglei f45dc9541b use mmcv.ProgressBar to replace tqdm 2020-07-05 21:57:42 +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
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