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
a24a9f6faa
[Fix] Build compatible with low pytorch versions ( #301 )
...
* add version compatible for torchscript
* doc
* doc again
* fix lint
* fix lint isort
2021-06-14 23:25:35 +08:00
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
c2f01e0dcd
[Feature] Add torchscript deployment ( #279 )
...
* add torchscript deploy
* fix lint
* add check and delete \
2021-06-12 21:50:48 +08:00
dbddde52ef
[Feature] TensorRT test tools. ( #284 )
...
* first commit
* update resnext result
* update docs
* update docstring
2021-06-12 21:47:10 +08:00
e2507cae97
Add mmcls2torchserve ( #292 )
...
* Add mmcls2torchserve
* Update docs
2021-06-12 21:45:45 +08:00
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
4ca21c7d03
[WIP] Refactoring weights initialization ( #270 )
...
* [WIP] Refactoring weights initialization
* fix lint and constant init cfg
* fix pretrained bug
* fix typo
* fix isort
* revise model utils
2021-06-10 10:54:34 +08:00
5066e32306
dump config before training ( #282 )
2021-06-04 16:26:13 +08:00
0a92a81198
Update analyze_results.py ( #237 )
...
correct spelling mistake(sucess->success)
2021-05-10 15:04:08 +08:00
01d2849b76
[Feature]: add onnxruntime test tool ( #212 )
...
* [draft] add onnxruntime accuruacy verification
* fix a bug
* update code
* fix lint
* fix lint
* update code and doc
* update doc
* update code
* update code
* updata doc and updata code
* update doc and fix some bug
* update doc
* update doc
* update doc
* update doc
* update doc
* update doc
* fix bug
* update doc
* update code
* move CUDAExecutionProvider to first place
* update resnext accuracy
* update doc
Co-authored-by: maningsheng <maningsheng@sensetime.com>
2021-04-26 13:57:08 +08:00
9be435846c
Support training on CPU ( #219 )
...
* draft
* add parameter for training tools
* Update .pre-commit-config.yaml
2021-04-26 13:56:45 +08:00
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
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
2bd28435cf
Add simplify option in pytorch2onnx.py and Update doc ( #200 )
...
* add simplify option in pytorch2onnx.py and update relevant doc
* update doc about pytorch2onnx
* update code
* update docs
2021-04-09 13:05:25 +08:00
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
8eb845e718
support dynamic shape export to onnx ( #175 )
...
* support dynamic shape export to onnx
* update the code and reset the simple_test() of image.py
* fix dynamic_axes name is not a valid input/output name
2021-03-17 15:50:03 +08:00
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
f0d358e340
add onnx2tensorrt ( #153 )
2021-02-05 17:48:37 +08:00
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
a7d4739d2b
[Fix] Fix test.py ( #155 )
...
* modeify test.py
* modeify test.py
* revise according to comments
2021-01-31 15:53:43 +08:00
4f8fc9cbf3
[Feature] Add analyze_results.py to select imgs for success/fail ( #142 )
...
* add analyze_results.py to select imgs for success/fail
* minor change
* save imgs
2021-01-19 15:49:14 +08:00
f355f15485
Fix bug in gpu_ids in distributed training ( #107 )
...
* update gpu_ids in distributed training
* move linear scaling rule after getting correct gpu_ids
* Remove support for autoscale_lr
2020-12-09 16:28:03 +08:00
e75f2b7c35
add argument to override configs ( #91 )
2020-11-20 13:50:28 +08:00
edffbcc486
Add pytorch2onnx ( #20 )
...
* add pytorch2onnx
* add pytorch2onnx API
2020-09-30 20:13:36 +08:00
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
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
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
91227e59cf
fix mobilenetv2 convert tools ( #34 )
...
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 14:05:34 +08:00
1d5f3d8c24
update shufflenet_v1
2020-07-12 00:06:55 +08:00
cf3a530c73
Add two useful tools from mmdet
2020-07-12 00:06:53 +08:00
fbc68bf266
Add classifiers, heads, necks and losses
2020-07-07 19:32:06 +08:00
eb24a94b68
add mobilenetv2 convert code
2020-07-05 21:59:22 +08:00
046aa5c003
add shufflenet_v2 convert code
2020-07-03 19:21:18 +08:00
a0d4f95725
Add missing mmcls.core
2020-05-27 11:37:16 +08:00
a99e6eb10d
init commit
2020-05-21 21:21:43 +08:00
Ma Zerun
AllentDan
WRH
q.yao
David de la Iglesia Castro
LXXXXR
Miao Zheng
Forsaken
QingChuanWS
agim-a
mzr1996
QingChuanWS
RunningLeon
Lei Yang
Jerry Jiarui XU
Xiaojie Li
lixiaojie
yanglei
chenkai