1. 07 Jan, 2021 1 commit
  2. 06 Jan, 2021 1 commit
    • q.yao's avatar
      [Feature]: Add custom operators support for TensorRT in mmcv (#686) · 0de9e149
      q.yao authored
      
      
      * 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: default avatarmaningsheng <maningsheng@sensetime.com>
      0de9e149
  3. 31 Dec, 2020 1 commit
    • RunningLeon's avatar
      [Feature] add onnx simplify (#751) · 83732ec2
      RunningLeon authored
      * [Enhance]: add onxx simplify
      
      * add simple doc
      
      * add unit test
      
      * update docstring
      
      * resolve some comment
      
      * add test dependency:onnxoptimizer
      
      * Fix onnxruntime register empty libpath
      
      * test onnxruntime version
      
      * set checker to false
      
      * skip test_simplify for torch<1.5.0
      83732ec2
  4. 29 Dec, 2020 1 commit
    • tangyanf's avatar
      [Feature]: add custom op roialign for onnxruntime (#724) · 741e52dd
      tangyanf authored
      
      
      * add ort custom op roialign
      
      * Update roi_align.py
      
      * Update test_onnx.py
      
      * Update test_onnx.py
      
      * Update test_onnx.py
      
      * Update test_onnx.py
      
      * Update onnxruntime_register.cpp
      
      * Update roiAlign.h
      
      * Update roiAlign.cpp
      
      * lint modification
      
      * update roiAlign.cpp
      
      * lint check
      
      * lint check
      
      * lint fix
      
      * lint fix
      
      * fix lint
      
      * add link to commit
      Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
      741e52dd
  5. 28 Dec, 2020 1 commit
  6. 25 Dec, 2020 2 commits
  7. 24 Dec, 2020 1 commit
  8. 23 Dec, 2020 1 commit
    • q.yao's avatar
      [Feature]: Add custom operators support for onnxruntime in mmcv (#612) · 94810f22
      q.yao authored
      
      
      * add onnx support to roi_align and roi_pool
      
      * add softnms ort support
      
      * fix for lint
      
      * format cpp code with clang-format:google
      
      * add new empty line to the end of head files in onnxruntime
      
      * update to pytorch1.7
      
      * add test of softnms to onnxruntime
      
      * fix for lint
      
      * remote print in ops/info.py
      
      * change import order, fix for flake8
      
      * fix include
      
      * add assert torch>=1.7.0
      
      * [doc]: add document for onnxruntime custom operator
      
      * update onnxruntime version to v1.5.1 for softnms
      
      * remove doc menu
      
      * Resolve lint for markdown
      
      * resolve naming style in onnxruntime_op.md
      
      * Use old cpp apis, optimize test_onnx.py
      
      * Fixing strings in tests/test_ops/test_onnx.py
      
      * code format with yapf
      
      * fix soft_nms parrot
      
      * add import in onnxruntime setup, avoid conflict
      
      * fix doc and add assert
      
      * change cpp guard
      Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
      94810f22
  9. 22 Dec, 2020 2 commits
  10. 20 Dec, 2020 1 commit
    • Wang Xinjiang's avatar
      Allow to replace nested tuple and list via options (#731) · 96ebfa65
      Wang Xinjiang authored
      * Allow to replace nested tuple and list via options
      
      * Add comments
      
      * Fix single nested items
      
      * Simplify the code
      
      * Simplify the code
      
      * Simplify the code
      
      * Simplify the code
      
      * Update docstring
      
      * Update docstring
      
      * Support quotation mark
      
      * modify docstring
      96ebfa65
  11. 18 Dec, 2020 1 commit
  12. 16 Dec, 2020 1 commit
  13. 15 Dec, 2020 1 commit
    • lml131's avatar
      Lml/jit decorator (#673) · 9befc398
      lml131 authored
      * add jit decorator
      
      * add parrots_jit.py
      
      * modify test_parrots_jit.py
      
      * modify for lint
      
      * fix isort
      
      * skip test_parrots_jit.py when build without pytorch
      
      * try ci
      
      * rm log
      
      * fix double quote
      
      * modify for comments and use partial_shape instead of full_shape
      
      * fix for lint
      
      * small modify for parrots 0.9.0rc0
      
      * def skip no elena directly
      9befc398
  14. 13 Dec, 2020 2 commits
    • BigBigDream's avatar
      fix roi_align ci for parrots (#708) · 02920db2
      BigBigDream authored
      * fix roi_align ci for parrots
      
      * fix lint
      02920db2
    • Rui Xu's avatar
      [Refactor] remove the c implementation of flow_warp (#707) · b7136e39
      Rui Xu authored
      * remove flow-warp-c
      
      * remove flow warp in setup.py and imports
      
      * fix floor and ceil bug
      
      * fix broadcast bug
      
      * add warnings and assertions
      
      * fix bilinear bug
      
      * pass unit test
      
      * pass unit test
      
      * pass unit test
      
      * pass unit test
      
      * fix value bug
      
      * fix lint
      
      * fix lint
      
      * add mk lint
      
      * update docs
      
      * update docs
      
      * fix bug in imports
      
      * fix bug in setup.py
      b7136e39
  15. 12 Dec, 2020 1 commit
  16. 09 Dec, 2020 1 commit
  17. 08 Dec, 2020 1 commit
  18. 25 Nov, 2020 1 commit
    • BigBigDream's avatar
      Add box_iou_rotated, ml_nms_rotated and nms_rotated (#625) · a9787648
      BigBigDream authored
      
      
      * add box_iou_rotated, ml_nms_rotated and nms_rotated
      
      * fix lint
      
      * fix lint
      
      * fix .py lint
      
      * fix cpp lint
      
      * add newline at the end
      
      * add new line
      
      * fix unittest
      
      * config google style
      
      * fix lint
      
      * lint
      
      * lint
      
      * yapf
      
      * update
      
      * fix lint
      
      * fix lint
      
      * fix lint
      
      * fix
      
      * fix format
      
      * fix format
      
      * add modified from
      
      * add docstring and update others
      
      * update docstring
      
      * update docstring
      
      * update
      
      * fix bug
      
      * fix bug
      
      * fix bug
      Co-authored-by: default avatarCao Yuhang <yhcao6@gmail.com>
      a9787648
  19. 23 Nov, 2020 1 commit
  20. 19 Nov, 2020 1 commit
  21. 18 Nov, 2020 2 commits
  22. 15 Nov, 2020 2 commits
  23. 14 Nov, 2020 1 commit
  24. 11 Nov, 2020 2 commits
    • yamengxi's avatar
      [Feature] Add CLAHE method (#647) · c6c230df
      yamengxi authored
      * add CLAHE
      
      * add CLAHE
      
      * restore
      
      * Add docstring
      
      * modify docstring
      
      * modify CLAHE to clahe
      
      * fix syntax error
      
      * simplify assert
      
      * simplify assert
      
      * add assert test
      
      * fix unittest bug
      
      * fix syntax bug
      
      * fix assert bug
      c6c230df
    • yamengxi's avatar
      [Feature] Add LUT transform (#646) · d9ef9dab
      yamengxi authored
      * add lut_transform
      
      * restore
      
      * fix test bug
      
      * add test
      
      * Fix syntax bug
      
      * Delete test id
      
      * modfiy docstrinng
      
      * simplify assert
      d9ef9dab
  25. 06 Nov, 2020 1 commit
  26. 05 Nov, 2020 1 commit
    • Kuro Latency's avatar
      Update lr_updater.py (#574) · bcf85026
      Kuro Latency authored
      * Update lr_updater.py
      
      since epoch/iteration in  runner starts with 0, we shouldn't leave the latter iteration to former (12th epoch for example, with first period equal to 12) period.
      
      * Update lr_updater.py
      
      * Update test_hooks.py
      bcf85026
  27. 31 Oct, 2020 1 commit
  28. 28 Oct, 2020 2 commits
  29. 26 Oct, 2020 1 commit
    • robin Han's avatar
      add unittest for onnx convert (#608) · 23b2bdbf
      robin Han authored
      * add unittest for onnx convert
      
      * build onnx and onnxruntime in CI
      
      * skip onnx op unit test while using CUDA
      
      * fix offset==0 case in NMS
      
      * remove tmp file used in test
      
      * delete tmp file before assert so that we can remove the tmp file anyway
      23b2bdbf
  30. 15 Oct, 2020 1 commit
  31. 06 Oct, 2020 1 commit
    • Wenwei Zhang's avatar
      Fix wrappers version comparison (#602) · 005c4087
      Wenwei Zhang authored
      * add version check in wrappers
      
      * fix assersion
      
      * use digital version for version comparison
      
      * fix unit tests
      
      * reformat
      
      * fall back to compare the first two version
      
      * fix unittest
      
      * fix unittest
      
      * fix unit test
      
      * clean unnecessary change
      005c4087
  32. 27 Sep, 2020 1 commit
    • GT9505's avatar
      register deconv in CONV_LAYERS (#582) · acee61d7
      GT9505 authored
      * register deconv in CONV_LAYERS
      
      * use ConvTranspose2d implemented in MMCV
      
      * remove repetitive register_module
      
      * update
      
      * add unittest for deconv
      acee61d7
  33. 25 Sep, 2020 1 commit
    • Wenwei Zhang's avatar
      Support to specify LR of DCN's conv_offset (#344) · ed2887bb
      Wenwei Zhang authored
      * Support to specify LR of DCN's conv_offset
      
      * Resolve comments & add unit test
      
      * Resolve formats
      
      * Fix CI for DCN
      
      * Mock DCN when cpu only
      
      * Use mock for cpu testing
      
      * Fix docstring and support ModulatedDCN
      
      * set offset_lr_mult as dcn's arguments, link CU-49u01p
      
      * fix lr bug
      
      * fall back to set LR in constructor
      
      * resolve comments
      ed2887bb