1. 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
  2. 25 Feb, 2021 1 commit
  3. 24 Feb, 2021 2 commits
  4. 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 t...
      1ee5315e
  5. 22 Feb, 2021 1 commit
    • ftbabi's avatar
      Add GELU activation function (#843) · 8735815a
      ftbabi authored
      * 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
      8735815a
  6. 21 Feb, 2021 2 commits
  7. 20 Feb, 2021 3 commits
  8. 18 Feb, 2021 2 commits
  9. 10 Feb, 2021 1 commit
  10. 07 Feb, 2021 4 commits
  11. 05 Feb, 2021 2 commits
  12. 04 Feb, 2021 2 commits
  13. 02 Feb, 2021 2 commits
  14. 01 Feb, 2021 1 commit
  15. 28 Jan, 2021 2 commits
    • RunningLeon's avatar
      [Doc]: add a short document for TensorRT plugins in mmcv (#797) · 5afca135
      RunningLeon authored
      * add trt doc
      
      * add notes
      
      * add sphinx_markdown_tables
      
      * resolve comments
      
      * resolve comments
      5afca135
    • Haian Huang(深度眸)'s avatar
      Refactor _load_checkpoint fn (#790) · 96892bdc
      Haian Huang(深度眸) authored
      * 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
      96892bdc
  16. 27 Jan, 2021 1 commit
  17. 21 Jan, 2021 1 commit
  18. 20 Jan, 2021 2 commits
  19. 14 Jan, 2021 1 commit
  20. 09 Jan, 2021 1 commit
  21. 08 Jan, 2021 5 commits
  22. 07 Jan, 2021 1 commit