1. 10 May, 2021 2 commits
  2. 01 May, 2021 1 commit
    • v-qjqs's avatar
      [Feature]: Support corner_pool related custom operators for onnxruntime in mmcv (#997) · db6b0542
      v-qjqs authored
      * supports for onnxruntime custom op `mmcv::MMCVTopPool`
      
      * supports for onnxruntime custom op `mmcv::MMCVCornerPool`, involving TopPool, BottomPool, LeftPool and RightPool
      
      * add unittest for corner_pool
      
      * supports mmcv::CornerPool without memcpy
      
      * add docs for mmcv::CornerPool
      
      * re-add docs for mmcv::CornerPool
      
      * fix output dtype doc
      
      * reformat
      
      * format with pre-commit
      
      * format
      
      * fix lint error, by using google clang-format style for c/c++
      db6b0542
  3. 27 Apr, 2021 2 commits
  4. 25 Apr, 2021 1 commit
  5. 23 Apr, 2021 1 commit
    • ZhangShilong's avatar
      [Feature]: support Multi-Scale-DeformAttention in deformable-detr (#878) · 54a7ebb4
      ZhangShilong authored
      * add c++ ms_deform_atten
      
      * fix cpp lint
      
      * fix cpp lint
      
      * clang format
      
      * remove cmakefile
      
      * google style
      
      * clang-format precommit
      
      * use clang-format-lint-action
      
      * add transformer base class
      
      * add merge
      
      * add docstr
      
      * add pyargs
      
      * fix according to commments
      
      * resiger module
      
      * change to use basemodule
      
      * add _ between build function
      
      * split the name
      
      * fix according to comments
      
      * fix lint and fix unitest
      
      * fix cpp lint
      
      * fix bug of deformdetr_atten
      
      * fix drop out
      
      * fix residual
      
      * use CUDA_1D_KERNEL_LOOP
      54a7ebb4
  6. 20 Apr, 2021 1 commit
  7. 19 Apr, 2021 1 commit
    • Wang Xinjiang's avatar
      [Feature]: Add Rotated ROI align op for pytorch (cpu&cuda), parrots (cpu&cuda)... · ee041cec
      Wang Xinjiang authored
      
      [Feature]: Add Rotated ROI align op for pytorch (cpu&cuda), parrots (cpu&cuda) and onnxruntime (cpu) (#933)
      
      * add roi_align_rotated
      
      * code format
      
      * Add align key to roi align rotated
      
      * Add clockwise for rotated roi align
      
      * fix bugs in onnx export
      
      * Add docstring for RoIAlignRotated
      
      * remove cuda unittest
      
      * Reformat c++ code
      
      * add onnx roi align rotated file
      
      * fix unittest
      
      * Add  cpu and float64 of cuda support for parrots
      
      * code format
      
      * Add unified header to roi align rotated
      Co-authored-by: default avatarluopeichao <luopeichao@sensetime.com>
      ee041cec
  8. 12 Apr, 2021 2 commits
  9. 10 Apr, 2021 3 commits
    • Jerry Jiarui XU's avatar
      add model registry (#760) · 375605fb
      Jerry Jiarui XU authored
      * add model registry
      
      * fixed infer scoep
      
      * fixed build func
      
      * add docstring
      
      * add md
      
      * support multi level
      
      * clean comments
      
      * add docs
      
      * fixed parent
      
      * add more doc
      
      * add value error, add docstring
      
      * fixed docs
      
      * change to local/global search
      
      * resolve comments
      
      * fixed test
      
      * update some docstring
      
      * update docs (minior)
      
      * update docs
      
      * update docs
      375605fb
    • Miao Zheng's avatar
      [Refactoring] Revise init_weight in BaseModule (#905) · 47825b19
      Miao Zheng authored
      * [Refactoring] Add deprecated API warning
      
      * revise test
      
      * fix lint
      
      * fix lint
      47825b19
    • Miao Zheng's avatar
      [Fix] OneCycleLrUpdaterHook adds total_steps (#929) · 89efc607
      Miao Zheng authored
      * [Fix] OneCycleLrUpdaterHook interface
      
      * revise according to comments
      
      * revise according to comments
      
      * add test
      
      * fix lint
      
      * revise according to comments
      
      * minors
      
      * add pytest param
      
      * fix lint
      
      * ci
      89efc607
  10. 09 Apr, 2021 2 commits
  11. 08 Apr, 2021 1 commit
  12. 06 Apr, 2021 1 commit
    • tangyanf's avatar
      [feature]:add onnxruntime custom op grid_sample (#916) · 83cf25b2
      tangyanf authored
      * add onnxruntime custom op grid_sample
      
      * update code
      
      * update code
      
      * update code
      
      * update code
      
      * update code
      
      * update code
      
      * update code
      
      * update code
      
      * update code
      
      * update code
      
      * update code
      
      * update code
      
      * update code
      
      * update code
      83cf25b2
  13. 02 Apr, 2021 1 commit
  14. 30 Mar, 2021 1 commit
    • q.yao's avatar
      [Feature] : Add grid_sampler TensorRT Plugin (#865) · 9d80f56a
      q.yao authored
      * add grid sample trt support
      
      * fix align_corners=True
      
      * change 'intep' to 'interp', fix bugs of grid sampler, better test script
      
      * remove unused import
      
      * move source comment to the top of trt_grid_sampler_kernel.cu
      9d80f56a
  15. 26 Mar, 2021 2 commits
  16. 24 Mar, 2021 1 commit
  17. 21 Mar, 2021 1 commit
  18. 20 Mar, 2021 1 commit
  19. 19 Mar, 2021 1 commit
  20. 17 Mar, 2021 1 commit
  21. 12 Mar, 2021 1 commit
  22. 11 Mar, 2021 1 commit
  23. 09 Mar, 2021 1 commit
  24. 08 Mar, 2021 2 commits
  25. 03 Mar, 2021 3 commits
    • LXXXXR's avatar
      27e9ede5
    • sshuair's avatar
      add HTTP backend (#860) · 6719cde4
      sshuair authored
      * add HTTP backend
      
      * add HTTPbackend
      
      * fix linting
      
      * fix linting
      
      * add docs attributes http
      
      * fix format
      
      * add real http link test for http backend
      6719cde4
    • Ycr's avatar
      [Feature] Support revise_keys in load_checkpoint(). (#829) · e076c8b0
      Ycr authored
      * 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.
      e076c8b0
  26. 26 Feb, 2021 2 commits
    • z55250825's avatar
      Add new parrots extension implementation for all ops (#794) · 48d99025
      z55250825 authored
      * 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
      48d99025
    • RunningLeon's avatar
      Fix pytorch2onnx failed for interpolate op with PyTorch==1.6.0(mmdet#4646) (#848) · 72e4cc12
      RunningLeon authored
      * Fix pytorch2onnx for yolov3 with torch==1.6.0
      
      * update and add test for F.interpolate
      
      * add test_onnx.py with build_cuda
      72e4cc12
  27. 25 Feb, 2021 1 commit
  28. 23 Feb, 2021 1 commit
    • q.yao's avatar
      [Feature] : Add NonMaxSuppression TensorRT Plugin (#787) · 1ee5315e
      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
      
      * 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: default avatarmaningsheng <maningsheng@sensetime.com>
      1ee5315e
  29. 20 Feb, 2021 1 commit