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

14 Commits (5c4da16084671d943855a90977e695bde59db7a3)

Author SHA1 Message Date
Ma Zerun aad796ae6f
Modify test tools and add some new tools (#322) 
* Refactor tools folder structure.

* Modify tools/test.py and add eval_metric.py to analysis test output.

* Add new tools `analyze_logs.py` and `print_config.py`.

* Add comment for analysis_tools functions.
 2021-06-30 20:51:16 +08:00
WRH b99bd4fa88
Fix bug for CPU training (#286) 
* remove MMDataParallel when using cpu

* support cpu testing

* fix lint
 2021-06-12 22:26:33 +08:00
LXXXXR 6de635a81c
[Bug] Missing test data when num_imgs can not be evenly divided by num_gpus (#299) 
* fix bug in test

* remove unneccesary code
 2021-06-11 15:53:23 +08:00
agim-a 3affc481c8
[Fix] check for CLASSES in checkpoint meta (#207) 
- check for CLASSES in checkpoint meta when key meta does not exists
 2021-04-15 22:19:23 +08:00
LXXXXR e76c5a368d
[Feature] Support fp16 training (#178) 
* change mmcls fp16 to mmcv hook

* support fp16

* clean unnessary stuff
 2021-03-17 15:53:55 +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 a7d4739d2b
[Fix] Fix test.py (#155) 
* modeify test.py

* modeify test.py

* revise according to comments
 2021-01-31 15:53:43 +08:00
LXXXXR e75f2b7c35
add argument to override configs (#91) 2020-11-20 13:50:28 +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
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
yanglei fbc68bf266 Add classifiers, heads, necks and losses 2020-07-07 19:32: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