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

41 Commits (31a6a362ba33b744797d908cbe513873c0ce4a6c)

Author SHA1 Message Date
mzr1996 b7b520881f
Update CONTRIBUTING.md according to mmcv (#210) 
* Update CONTRIBUTING.md according to mmcv

* Docstring formatting by docformatter

* Update openmmlab website.
 2021-04-14 21:22:37 +08:00
LXXXXR 5195932952
[Feature] Support random augmentation (#201) 
* support random augmentation

* minor fix on posterize

* minor fix on posterize

* minor fix on cutout

* minor fix on cutout

* fix bug in solarize add

* revised according to comments
 2021-04-09 14:02:50 +08:00
LXXXXR 4d1fb1a662
[Feature] ColorJitter and Lighting (#190) 
* add configs

* remove config

* add color jitter and lighting

* revised according to comments
 2021-04-02 19:23:39 +08:00
LXXXXR 93cd960466
[Feature] Support AutoAug, AutoContrast, Equalize, Contrast, Brightness and Sharpness (#179) 
* add AutoContrast, Equalize, Contrast, Brightness and Sharpness pipelines

* add ImageNetPolicy

* add configs

* add unittest

* remove config

* rerun CI

* rerun CI

* [Fix] Update pip install mmcv command in ci (#187)

* update pip install mmcv command in ci

* update pip install mmcv command in ci

* fix ci

* fix ci
 2021-03-30 15:38:55 +08:00
LXXXXR f3b9380cbc
[Feature] Support solarize and posterize pipelines (#172) 
* support Solarize and Posterize

* revised according to comments

* fix conflicts
 2021-03-09 19:45:13 +08:00
LXXXXR 9614787fc4
[Feature] Add color pipeline (#171) 
* add ColorTransform pipeline

* fix docstring

* minor change

* revised according to comments
 2021-03-09 19:28:50 +08:00
LXXXXR c8033ece8e
add invert pipeline (#168) 2021-03-02 16:46:57 +08:00
LXXXXR b1fa298a66
add Rotate pipeline (#167) 2021-03-01 21:49:07 +08:00
LXXXXR bc50a7927b
reslove conflicts (#165) 2021-02-26 11:48:13 +08:00
LXXXXR 8c11c01fdb
[Feature] Add shear pipeline (#163) 
* half-done auto_augmentation

* remove auto_augcode and support shear pipeline

* fix typo

* fix typo

* use a non-square toy data instead
 2021-02-25 16:00:46 +08:00
David de la Iglesia Castro fb11a23cfe
Show results in test api (#162) 
* Initial support show in single_gpu_test

* Add result_show

* Add ori_filename meta_key

* test ori_filename

* Add show, show-dir, show-options
 2021-02-25 16:00:22 +08:00
LXXXXR ddc2a14177
[Feature] Add --eval-options in test.py (#158) 
* add --eval-options in test.py

* fix typo

* revise according to commnets
 2021-02-05 17:46:43 +08:00
LXXXXR 3e5a9513be
[Bug] Fix bug in Collect (#149) 
* fix bug in Collect

* add metakeys
 2021-01-25 20:29:28 +08:00
LXXXXR fc82c31b2f
fix RandomCrop and RandomResizedCrop (#151) 2021-01-25 20:28:16 +08:00
LXXXXR 07bb15e5fd
[Feature] Add heads and config for multilabel task (#145) 
* resolve conflicts
add heads and config for multilabel tasks

* minor change

* remove evaluating mAP in head

* add baseline config

* add configs

* reserve only one config

* minor change

* fix minor bug

* minor change

* minor change

* add unittests and fix docstrings
 2021-01-25 18:10:14 +08:00
LXXXXR 13c1210741
[Feature] Add thrs in eval_metrics (#146) 
* support thr

* replace thrs with thr

* fix docstring

* minor change

* revise according to comments

* revised according to comments

* revise according to comments

* rewrite basedataset.evaluate to avoid duplicate calculation

* minor change

* change thr to thrs

* add more unit test
 2021-01-25 17:54:22 +08:00
LXXXXR 8e990b5654
[Feature] Support support and class-wise evaluation results (#143) 
* support support, support class-wise evaluation results and move eval_metrics.py

* Fix docstring

* change average to be non-optional

* revise according to comments

* add more unittest
 2021-01-19 16:42:16 +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
LXXXXR c07e60dd7d
[Feature] Support PASCAL VOC 2007 dataset for multilabel task (#134) 
* support voc

* minor change
 2021-01-11 11:08:59 +08:00
LXXXXR 92438da12a
fix bug (#108) 2020-12-04 11:41:49 +08:00
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
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
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
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
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
anthracene fef5202257
Fix crash in CenterCrop transform when image is greyscale (#40) 2020-08-31 00:04:39 +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 e29882c8af fix old register 2020-07-12 00:06:56 +08:00
yanglei 6968ad5b3b Add RandomFlip 2020-07-12 00:06:56 +08:00
lixiaojie f6260b33bf Add RandomCrop, RandomResizedCrop, RandomGrayscale, impad 2020-07-12 00:06:55 +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 f45dc9541b use mmcv.ProgressBar to replace tqdm 2020-07-05 21:57:42 +08:00
yanglei ba789da5c8 Add commonly used datasets 2020-07-01 16:09:06 +08:00
yanglei a0d4f95725 Add missing mmcls.core 2020-05-27 11:37:16 +08:00
chenkai a99e6eb10d init commit 2020-05-21 21:21:43 +08:00