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

118 Commits (5b266d9e7c49044d1fcf15d046df3c06af3ff505)

Author SHA1 Message Date
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