b1e91f256b
fix bug in recall and precision ( #112 )
2020-12-09 16:27:42 +08:00
92438da12a
fix bug ( #108 )
2020-12-04 11:41:49 +08:00
400074c321
Add markdown lint in pre-commit hook ( #106 )
...
* add mdlint
* remove duplicate check
* modify md files to pass linting check
* change mmcv download url
* remove torch1.4.0 version check
* remove torch1.4.0 version check
2020-12-02 19:42:45 +08:00
6f7698cb7c
Update accuracy.py ( #104 )
...
Co-authored-by: Ülkü Tuncer Küçüktaş <UlkuTuncerKucuktas@users.noreply.github.com>
2020-12-01 14:25:18 +08:00
5de2d2c0e9
Change to use mmcv.utils.collect_env to avoid duplicate codes ( #103 )
2020-12-01 14:24:08 +08:00
374b13c419
replace urls in model zoo ( #102 )
2020-12-01 14:23:32 +08:00
761c39b6b9
Add config link in model_zoo.md ( #99 )
...
* add config link in model_zoo.md
* change model download path
2020-11-26 15:47:22 +08:00
44bbc71e14
Fix bug and optimize MNIST config ( #98 )
...
* add simple_test to ClsHead
* optimize lenet training config
* recover path setting
2020-11-26 15:27:04 +08:00
c0e7512969
support options to override configs in inference.py ( #96 )
2020-11-25 16:14:01 +08:00
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
e75f2b7c35
add argument to override configs ( #91 )
2020-11-20 13:50:28 +08:00
149ee3a30d
support img input as ndarray in inference.py ( #87 )
...
* support img input as ndarray
* revise according to comments
2020-11-19 18:58:25 +08:00
fe45b241c5
Do not install mmcv through requirements ( #90 )
...
* Delete the mmcv requirement to support mmcv-full installed environment
* install mmcv in CI
* fix mmcv version error
Co-authored-by: Xinran <lixinran@sensetime.com>
2020-11-19 00:00:12 +08:00
7636409b3b
add get_class in base_dataset ( #85 )
...
* modify base_dataset
* revise according to the comments
2020-11-12 14:22:02 +08:00
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
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
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
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
f7a916f309
fix pypi release error ( #64 )
2020-10-11 00:26:34 +08:00
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
c00ee6a876
workflow: modify build, add deploy ( #61 )
...
* workflow: modify build, add deploy
* fix type: mmseg -> mmcls
* simplify build by delete cuda related settings
* re-sort requirements by pre-commit hook
* build: support more pytorch versions
* build: fix torchvision version bug
* del useless test imports
* merge upstream
* build: del docstring checking to pass coverage check
2020-10-10 22:15:50 +08:00
24fd4fb627
Visualize results on image demo ( #58 )
...
* visualize results on image demo
* add matplotlib in requirements
2020-10-10 16:33:27 +08:00
52d8ea4621
Merge pull request #53 from daavoo/fix-configs
...
fix outdated configs
2020-10-01 23:05:56 +08:00
b577594d3d
fix outdated configs
2020-10-01 16:15:14 +02: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
b6774b1224
Update imagenet data setup in getting_started document ( #48 )
2020-09-30 01:06:23 +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
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
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
6b0b3da92a
Add note about ground-truth labels range ( #44 )
...
* Add note about ground-truth labels range
* Fix
2020-09-09 16:21:25 +08:00
fef5202257
Fix crash in CenterCrop transform when image is greyscale ( #40 )
2020-08-31 00:04:39 +08:00
4b46fd6dc7
Add CIFAR10 configs and models ( #38 )
...
* rename cifar configs
* add cifar10 configs
* fix linting
* add params, flops and accuracy in docs
* add cifar10 pretrained models and logs
* use oss-accelerate url
* del unused cifar10 configs
2020-08-26 16:34:15 +08:00
b2fc837e84
Rename configs with samples_per_batch x gpus ( #37 )
...
* rename configs with samples_per_batch x gpus
* revise docs accordingly
2020-08-23 23:53:51 +08:00
91227e59cf
fix mobilenetv2 convert tools ( #34 )
...
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 14:05:34 +08:00
59abed3c39
Update pretrained model ( #33 )
...
* update model_zoo
* update performance
* update shufflenet v2
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 14:04:57 +08:00
18ac0f0fa3
Update shufflenet config ( #31 )
...
* update shufflenet config
* remove old config
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 10:46:39 +08:00
414ede4ca4
fix coslr ( #32 )
...
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 10:45:41 +08:00
608cea2469
fix ( #30 )
...
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 09:50:14 +08:00
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
408d92bcbe
revise docstr of alexnet to correct input_size ( #26 )
2020-08-12 16:14:19 +08:00
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
3e2d3d726a
Merge pull request #21 from yl-1993/dev/resize_backend
...
Add PIL backend for resize transform
2020-08-07 21:50:51 +08:00
0009591aa3
add unittest for backends
2020-08-05 16:53:35 +08:00
a0c814fecd
add doc for backend
2020-08-05 15:35:40 +08:00
360ae80591
rename detectors to classifiers ( #19 )
2020-08-05 00:43:13 +08:00
7d1edd03c6
revise a part of configs to pil resize
2020-08-03 12:54:58 +08:00
8af1537569
add pil resize configs
2020-08-03 12:54:08 +08:00
34efbc8a9d
add pil backend for resize
2020-08-03 12:52:35 +08:00
557a364d25
replace urlretrieve with urlopen ( #13 )
2020-07-27 22:20:03 +08:00
LXXXXR
Ülkü Tuncer Küçüktaş
WRH
Wenwei Zhang
Lei Yang
David de la Iglesia Castro
Y. Xiong
Mansi Mane
Jerry Jiarui XU
anthracene
Xiaojie Li