Commit Graph

686 Commits (5a99f587ed09797ae6720d33888c753ad88f178a)
 

Author SHA1 Message Date
RunningLeon 3dd98daaa2
fix test unit of nms and batched_nms for tensorrt (#872) 2021-03-08 16:57:19 +08:00
LXXXXR ce10425d2f
[Feature] Add adjust sharpness (#864)
* add adjust sharpness

* revised according to commnts

* fix bug

* fix docstring

* revise according to comments

* fix blank lines

* revised according to comments
2021-03-08 10:23:49 +08:00
LXXXXR 27e9ede5b9
[Fix]: fix bugs in some photometric transforms (#863) 2021-03-03 18:58:15 +08:00
sshuair 6719cde43a
add HTTP backend (#860)
* add HTTP backend

* add HTTPbackend

* fix linting

* fix linting

* add docs attributes http

* fix format

* add real http link test for http backend
2021-03-03 16:23:04 +08:00
Ycr e076c8b042
[Feature] Support revise_keys in load_checkpoint(). (#829)
* Simplified the code.

* Improved chkpt compatibility.

* One may modify the checkpoint via adding keywords.

* Tiny.

* Following reviewer's suggestion.

* Added unit_test.

* Fixed.

* Modify the state_dict  with  construction.

* Added test.

* Modified。

* Mimimalised the modification.

* Added the docstring.

* Format.

* Improved.

* Tiny.

* Temp file.

* Added assertion.

* Doc string.

* Fixed.
2021-03-03 10:59:11 +08:00
Jerry Jiarui XU 34b552b881
[Feature] Support save last checkpoint (#853)
* [Feature] Support save last checkpoint

* move to before_run, update doc

* move to after train

* add comments
2021-02-28 12:50:17 +08:00
Jintao Lin e5eaf2a729
update video/io.py (#856) 2021-02-27 10:28:39 +08:00
z55250825 48d9902585
Add new parrots extension implementation for all ops (#794)
* delete all parrots file
add bbox_overlaps new parrots op impl

* support first new impl parrts op (bbox_overlaps)(success test)

* add box_iou_rotated op, test succeed

* add carafe and carafe_naive op, test succeed (one parrots bug need fix)

* add cc_attention op, test success

* add corner_pool op, test success

* add parrots op deform_conv, test success

* add deform_roi_pool op, test success (but has question)

* add focal loss op, test success (gradcheck)

* add masked_conv2d op, test success

* add modulated_deform_conv op, test success

* add nms and nms_rotated op, test success

* add psamask op, test success

* add roi_align op, test_success

* add roi_pool op, test success

* add sync_bn op, test success

* add tin_shift op, test success

* fix test_deform_roi_pool, add parrots test

* skip test_onnx because parrots does not support onnx

* fix c++ lint

* fix python lint

* fix python lint
2021-02-26 19:05:25 +08:00
RunningLeon 72e4cc12bd
Fix pytorch2onnx failed for interpolate op with PyTorch==1.6.0(mmdet#4646) (#848)
* Fix pytorch2onnx for yolov3 with torch==1.6.0

* update and add test for F.interpolate

* add test_onnx.py with build_cuda
2021-02-26 10:49:12 +08:00
Haian Huang(深度眸) 58a8483352
Fix the iter error when the number of GPUs is different during resume (#844)
* Fix the iter error when the number of GPUs is different during resume

* Add fromstring and unit test

* Remove is_pretty_text

* Fix comment

* Add log info

* Add py format check

* Remove SyntaxError check
2021-02-25 16:40:06 +08:00
Miao Zheng ba30d98a7b
Add docstring for DistSamplerSeedHook (#850) 2021-02-24 10:07:46 +08:00
Jintao Lin 823b62ed94
Fix typo in README (#852) 2021-02-24 10:07:18 +08:00
q.yao 1ee5315ea2
[Feature] : Add NonMaxSuppression TensorRT Plugin (#787)
* start trt plugin prototype

* Add test module, modify roialign convertor

* finish roi_align trt plugin

* fix conflict of RoiAlign and MMCVRoiAlign

* fix for lint

* fix test tensorrt module

* test_tensorrt move import to test func

* add except error type

* add tensorrt to setup.cfg

* code format with yapf

* fix for clang-format

* move tensorrt_utils to mmcv/tensorrt, add comments, better test module

* fix line endings, docformatter

* isort init, remove trailing whitespace

* add except type

* fix setup.py

* put import extension inside trt setup

* change c++ guard, update pytest script, better setup, etc

* sort import with isort

* sort import with isort

* move init of plugin lib to init_plugins.py

* add scatternd, nms plugin (WIP)

* fix bugs of trt_nms

* add trt nms test module

* fix bugs of scatternd

* code optimize, add comment about nms kernel

* fix transform_if bug of trt_nms_kernel

* fix struct name

* default nms offset=0, fix bugs of batched input

* format with clang-format

* onnx preprocess

* much better nms implementation, no need to transfer memory between host and device

* update preprocess_onnx

* parse constant tensor from initializer in preprocess_onnx

* update nms

* remove unnecessary codes

* workspace aligned address

* format trt_plugin_helper.hpp

* fix index memory bugs

* set alignment to 16 by default

* fix lint

* fix nms offset

* fix bugs of preprocess onnx

* update test for nms

* tensorrt only accept int32, not int64

* update nms comments

* fix indexing for scores in nms

* update trt temp

* make trt-nms compatiable to #803

* fix lint

* add docstring to trt_nms_kernel.cuda, add description to preprocess_onnx

* add comment to score indexing

* fix bugs of max output boxes

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2021-02-23 15:09:49 +08:00
ftbabi 8735815a0d
Add GELU activation function (#843)
* Support GELU activation function

* Support different torch version

* Fix bug in importing TORCH_VERSION

* Wrap GELU module for PyTorch 1.3.x

* Wrap GELU module for PyTorch 1.3.x
2021-02-22 10:55:42 +08:00
Kai Chen b2ffab1306
[feature] Add readme of chinese version (#846)
* add readme of chinese version

* add contribution guide
2021-02-21 21:24:53 +08:00
ftbabi 9637bcff7c
Fix document in cnn.md (#841) 2021-02-21 11:13:39 +08:00
Y. Xiong ca47ae1f87
Bump version to v1.2.7 (#842) 2021-02-20 17:57:05 +08:00
Miao Zheng 72e13e6a0b
fix baserunner bug (#840)
* fix baserunner bug

* add unit test

* hooks type check
2021-02-20 12:46:40 +08:00
Miao Zheng 7fa78e7a1c
add register_itertimer_hook function (#838)
* add register_itertimer_hook function

* set default value

* revise minors

* revise according to comments

* fix according to comments

* update

* update
2021-02-20 10:46:31 +08:00
Zhiyuan Chen f75a88c297
fix no torch.no_grad decorator for val in iter_based_runner (#835) 2021-02-18 22:02:11 +08:00
Wang Xinjiang 3d229dbaf2
Add more error info in build_from_cfg (#834)
* Add more Error info in building from config

* Add more Error info in building from config

* change error info
2021-02-18 13:07:42 +08:00
lizz 1c2e665ad6
map_location for all (#826)
* map_location for all

* format

* hmm

* map_location

* back

* doc

* same
2021-02-10 12:23:39 +08:00
Miao Zheng 999f2d08b4
fix weight_init.py (#825)
* fix weight_init.py

* revise BaseInit args
2021-02-07 17:24:03 +08:00
ZhangShilong 4712db75d6
Fix bug of DCN (#819)
* fix bug of DCN

* fix clang=format
2021-02-07 13:05:27 +08:00
Miao Zheng a4c3702628
add initializers and BaseModule for unified parameter initialization (#780)
* add initializers and BaseModule for unified parameter initialization

* fix circle import

* bug fix

* add is_init flag in BaseModule

* fix docstring

* sort import and fix doc format

* fix bug

* fix docformat and double quote string

* fix import sort

* import sort

* sort import

* revise according to comments

* fix doc format

* revise according to comments

* revise import and fix typo

* polish code

* revise minors

* revice minors

* revise apply function

* revise bias initialization with probability

* add type test for bias_prob

* revise minors
2021-02-07 11:13:57 +08:00
Luting Wang 11b9264053
fix: imshow_bboxes fails with incontiguous img (#823) 2021-02-07 11:10:42 +08:00
lml131 4a694cc0ee
add PARROTS_JIT_OPTION (#816)
* code update

* add PARROTS_JIT_OPTION

* rm flow_warp_module

* fix for lint

* small fix
2021-02-05 19:58:33 +08:00
RunningLeon 5df759ad44
[Fix]: add header file for std::iota(#803) (#820)
* add header to std::iota

* fix lint

* fix lint for cpp
2021-02-05 12:52:19 +08:00
RunningLeon 459d728c47
[Fix]: parse scales of list in PyTorch==1.7.0 (#451 in MMPose) (#815)
* modify symbolic

* add comments
2021-02-04 15:47:19 +08:00
tangyanf be2616a0e2
add ort nms op (#803)
* add ort nms op

* fit lint check

* fix lint check

* update code

* fix lint check

* update code

* update code

* update code

* update code

* update code

* update code
2021-02-04 15:45:09 +08:00
ftbabi 6c57b88f5d
Fix document in io.md (#813) 2021-02-02 22:50:18 +08:00
Rui Xu cf2f14d06c
[Feature] Add file mode for logger (#808)
* add file mode for logger

* add more docs and comments
2021-02-02 22:49:24 +08:00
Jintao Lin 93980b4b91
[Docs] Polish CONTRIBUTING for all OpenMMLab projects (#807)
* polish CONTRIBUTING for all OpenMMLab projects

* polish some description
2021-02-01 15:38:03 +08:00
RunningLeon 5afca135f7
[Doc]: add a short document for TensorRT plugins in mmcv (#797)
* add trt doc

* add notes

* add sphinx_markdown_tables

* resolve comments

* resolve comments
2021-01-28 15:57:13 +08:00
Haian Huang(深度眸) 96892bdc42
Refactor _load_checkpoint fn (#790)
* Refactor _load_checkpoint fn

* Update _load_checkpoint fn

* Update docs str and add unit test

* Fix unit test

* Fix lint

* Add comment and Optimize function

* Fix docs str

* Update load_ckpt and fix doc str

* Update doc str and add sort unit test

* Update and fix unit test

* Fix unit test

* Update and add unit test

* Fix openmmlab prefix error
2021-01-28 15:09:19 +08:00
David de la Iglesia Castro 4450bd2ec2
Hande exif orientation with `pillow` backend in `imread` (#805)
* Add breaking test

* Handle exif orientation tag
2021-01-27 11:53:43 +08:00
Y. Xiong 9e601ca6ca
Bump version to v1.2.6 (#799) 2021-01-21 14:34:49 +08:00
lizz 00658e8cf3
Fix copy source file path (#796) 2021-01-20 15:54:53 +08:00
q.yao 2ab544fc29
[Feature] : Add ScatterND TensorRT Plugin (#786)
* add scatter plugin

* fix bugs of scatternd

* add trt scatternd plugin

* format code with clang-format

* add test for scatternd

* skip test_tensorrt in CI

* remove unused variable

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2021-01-20 11:15:07 +08:00
BigBigDream 8e3a801596
fix mmcv ci for parrots (#782)
* fix mmcv ci for parrots

* fix mmcv ci

* fix lint
2021-01-14 14:31:57 +08:00
Rui Xu f169fb5279
[Feature] Support not to register lr scheduler hook (#774)
* support no lr scheduler hook

* fix lint
2021-01-09 15:11:46 +08:00
Qiaofei Li 3d6032c6db
Add dataset classes name info to meta for saving ckpt (#776)
* add CLASSES to meta info

* Update checkpoint.py

* add unit test for CLASSES name

* clean up the tmp folder

* use tempfile to clean up temp folder
2021-01-08 16:40:43 +08:00
congee 905c9b43b8
Add common testing function of MM repos (#743)
* add testing function

add unittest for check_dict

add unittest for the function in testing

* polish docstring of testing.py

rename some function

* remove  in is_all_zeros

* modify the comment of check_dict

* modify the testing.py according to feedback

* add test about numpy for function dict_contains_subset

* applying unified style
2021-01-08 13:18:08 +08:00
Zhiyuan Chen daab369e99
Add torch.no_grad() decorator to the whole val workflow(#777) 2021-01-08 13:16:49 +08:00
RunningLeon 477f0c0a39
[Fix]: Set installation of onnx simplify's related packages to optional (#779)
* Fix onnx, onnxruntime, onnxoptimizer import

* Revert "Fix onnx, onnxruntime, onnxoptimizer import"

This reverts commit 820e3785a1.

* remove simplify from init file

* update test for onnx simplify

* update onnx doc
2021-01-08 13:15:37 +08:00
LXXXXR 51a0c23f60
[Feature] Support load checkpoint from ceph (#778)
* support load checkpoint using ceph

* minor change
2021-01-08 13:08:16 +08:00
Rui Xu 276883f120
[Feature] Allow register multi-name for a module simultaneously (#775)
* allow register multi-name for a module simultaneously

* add assertion for name type

* use isintance intead of is_str

* fix bug in unit test

* fix unit test
2021-01-07 22:27:20 +08:00
q.yao 0de9e149c0
[Feature]: Add custom operators support for TensorRT in mmcv (#686)
* start trt plugin prototype

* Add test module, modify roialign convertor

* finish roi_align trt plugin

* fix conflict of RoiAlign and MMCVRoiAlign

* fix for lint

* fix test tensorrt module

* test_tensorrt move import to test func

* add except error type

* add tensorrt to setup.cfg

* code format with yapf

* fix for clang-format

* move tensorrt_utils to mmcv/tensorrt, add comments, better test module

* fix line endings, docformatter

* isort init, remove trailing whitespace

* add except type

* fix setup.py

* put import extension inside trt setup

* change c++ guard, update pytest script, better setup, etc

* sort import with isort

* sort import with isort

* move init of plugin lib to init_plugins.py

* resolve format and add test dependency: tensorrt

* tensorrt should be installed from source not from pypi

* update naming style and input check

* resolve lint error

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2021-01-06 11:05:19 +08:00
Y. Xiong 643009e445
Bump version to v1.2.5 (#770) 2021-01-04 19:10:38 +08:00
Rui Xu 32bf3499bc
[Bug] Fix ddp bug when pytorch>=1.7 (#769)
* fix ddp bug when pytorch>=1.7

* fix ddp bug when pytorch>=1.7

* support pat

* fix docs

* use print_log instead of get_logger

Co-authored-by: nbei <631557085@qq.com>
2021-01-03 18:51:45 +08:00