1. 09 Jul, 2021 2 commits
  2. 03 Jul, 2021 1 commit
  3. 29 Jun, 2021 2 commits
  4. 25 Jun, 2021 4 commits
  5. 24 Jun, 2021 3 commits
  6. 23 Jun, 2021 1 commit
  7. 16 Jun, 2021 1 commit
  8. 11 Jun, 2021 2 commits
    • Shilong Zhang's avatar
      Refactor the baseclass related to transformer (#978) · e05fb560
      Shilong Zhang authored
      
      
      * minor changes
      
      * change to modulist
      
      * change to Sequential
      
      * replace dropout with attn_drop and proj_drop in MultiheadAttention
      
      * add operation_name for attn
      
      * add drop path and move all ffn args to ffncfgs
      
      * fix typo
      
      * fix a bug when use default value of ffn_cfgs
      
      * fix ffns
      
      * add deprecate warning
      
      * fix deprecate warning
      
      * change to pop kwargs
      
      * support register FFN of transformer
      
      * support batch first
      
      * fix batch first wapper
      
      * fix forward wapper
      
      * fix typo
      
      * fix lint
      
      * add unitest for transformer
      
      * fix unitest
      
      * fix equal
      
      * use allclose
      
      * fix comments
      
      * fix comments
      
      * change configdict to dict
      
      * move drop to a file
      
      * add comments for drop path
      
      * add noqa 501
      
      * move bnc wapper to MultiheadAttention
      
      * move bnc wapper to MultiheadAttention
      
      * use dep warning
      
      * resolve comments
      
      * add unitest:
      
      * rename residual to identity
      
      * revert runner
      
      * msda residual to identity
      
      * rename inp_identity to identity
      
      * fix name
      
      * fix transformer
      
      * remove key in msda
      
      * remove assert for key
      Co-authored-by: default avatarHIT-cwh <2892770585@qq.com>
      Co-authored-by: default avatarbkhuang <congee524@gmail.com>
      Co-authored-by: default avatarWenwei Zhang <40779233+ZwwWayne@users.noreply.github.com>
      e05fb560
    • Dmitry Sidnev's avatar
      [Feature] enable exporting to onnx for PointRend (#953) · a88d1d28
      Dmitry Sidnev authored
      
      
      * Fix export to onnx for PointRend
      
      * Fix codestyle
      
      * Fix codestyle
      
      * Fix type in docstring
      
      * Minor fix
      
      * Fix export with custom ops
      
      * Fix codestyle
      
      * Add tests for bilinear_grid_sample function
      
      * Remove redundant operation and rename variables
      
      * Fix bug in bilinear_grid_sample and update test
      
      * Fix getting batch size
      
      * skip torch==1.3.1
      
      * remove unused import
      
      * fix lint
      
      * support export with batch
      
      * fix dynamic clip
      
      * skip test for torch<1.5.0
      
      * Add docstrings and comments
      
      * Minor fix
      
      * Recover clipping code
      
      * Fix clamping in pytorch 1.7.0
      
      * Fix bilinear_grid_sampler
      
      * Minor fix
      Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
      a88d1d28
  9. 08 Jun, 2021 1 commit
  10. 01 Jun, 2021 1 commit
  11. 31 May, 2021 1 commit
    • SemyonBevzuk's avatar
      [Feature] NMS update (#957) · bf2c9fa8
      SemyonBevzuk authored
      
      
      * Add score_threshold and max_num to NMS
      
      * Fix codestyle
      
      * Fix codestyle
      
      * Fix inds in nms
      
      * Update nms docstring
      
      * Move score_threshold and max_num arguments
      
      * Fix args order in docstring
      
      * fix lint of c++ file
      
      * Remove torch.onnx.is_in_onnx_export() and add max_num to batched_nms for separate classes.
      
      * Rewrote max_num handling in NMSop.symbolic
      
      * Added processing max_output_boxes_per_class when exporting to TensorRT
      
      * Added score_threshold and max_num for NMS in test_onnx.py and test_tensorrt.py
      
      * Remove _is_value(max_num)
      
      * fix ci errors with torch==1.3.1
      
      * Update test_batched_nms in test_nms.py
      
      * Added tests for preprocess_onnx
      
      * Moved 'test_tensorrt_preprocess.py' and 'preprocess', updated 'remove_tmp_file'.
      
      * Update mmcv/tensorrt/__init__.py
      
      * Fix segfault torch==1.3.1 (remove onnx.checker.check_model)
      
      * Returned 'onnx.checker.check_model' with torch version check
      
      * Changed torch version from 1.3.1 to 1.4.0
      
      * update version check
      
      * remove check for onnx
      Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
      bf2c9fa8
  12. 25 May, 2021 3 commits
  13. 24 May, 2021 1 commit
  14. 23 May, 2021 4 commits
  15. 13 May, 2021 3 commits
  16. 12 May, 2021 1 commit
    • jeffreykuang's avatar
      Add pixel group and contour expand ops (#993) · 2623fbf2
      jeffreykuang authored
      * add pixel group ops
      
      * reformatting
      
      * formatting&rm auto
      
      * Add citation
      
      * Add contour expand
      
      * c++ linting
      
      * Add unit tests with Tensor
      
      * rm model.pth
      
      * rename
      
      * c++ linting
      
      * c++ linting
      
      * Rename variables
      2623fbf2
  17. 11 May, 2021 2 commits
  18. 10 May, 2021 2 commits
  19. 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
  20. 27 Apr, 2021 2 commits
  21. 25 Apr, 2021 1 commit
  22. 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