1. 22 Oct, 2021 2 commits
  2. 21 Oct, 2021 2 commits
  3. 15 Oct, 2021 3 commits
  4. 14 Oct, 2021 4 commits
  5. 13 Oct, 2021 2 commits
  6. 04 Oct, 2021 1 commit
  7. 25 Sep, 2021 1 commit
  8. 23 Sep, 2021 3 commits
  9. 09 Sep, 2021 1 commit
  10. 29 Aug, 2021 1 commit
    • Eugene Liu's avatar
      Add DCN and Modulated DCN CPU implementation (#1278) · e621e08d
      Eugene Liu authored
      * DCN cpu version
      
      * add modulated dcn cpu version
      
      * move deform_conv_shape_check to deform conv utils
      
      * add inline to deform_conv_shape_check
      
      * add tests
      
      * run linter
      
      * add newline at file end
      
      * run pre-commit against modulated deform conv cpp
      
      * update saconv test
      
      * run clang-format
      
      * remove cuda device inline
      
      * refactor dcn cuda/cpu functions
      
      * remove DCN util
      
      * remove DCN util hpp from all included files
      
      * Addressing PR comment by refactoring modulated-DCN
      
      * fix lint in cpp files
      e621e08d
  11. 23 Aug, 2021 1 commit
  12. 23 Jul, 2021 1 commit
    • Haodong Duan's avatar
      [Improvement] Improve digit_version & use it for version_checking (#1185) · ef48a473
      Haodong Duan authored
      * improve digit_version & use it for version_checking
      
      * more testing for digit_version
      
      * setuptools >= 50 is needed
      
      * fix CI
      
      * add debuging log
      
      * >= to ==
      
      * fix lint
      
      * remove
      
      * add failure case
      
      * replace
      
      * fix
      
      * consider TORCH_VERSION == 'parrots'
      
      * add unittest
      
      * digit_version do not deal with the case if 'parrots' in version name.
      ef48a473
  13. 20 Jul, 2021 1 commit
  14. 09 Jul, 2021 1 commit
  15. 29 Jun, 2021 1 commit
  16. 16 Jun, 2021 1 commit
  17. 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
  18. 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
      
      *...
      bf2c9fa8
  19. 25 May, 2021 3 commits
  20. 24 May, 2021 1 commit
  21. 23 May, 2021 1 commit
  22. 13 May, 2021 1 commit
  23. 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
  24. 10 May, 2021 1 commit
    • v-qjqs's avatar
      Support custom operators cummax and cummin for onnxruntime (#1010) · 934b549e
      v-qjqs authored
      * support custom op `mmcv::cummax` for onnxruntime in mmcv
      
      * fix clang-format lint error
      
      * support mmcv::cummin, reformat codes
      
      * fix merge from master
      
      * add docs for mmcv::cummax and mmcv::cummin
      
      * format doc
      
      * add assertion for torch version, when exporting `cummax` to onnx
      
      * add more comments for torch version
      
      * handle exporting to onnx in `soft_nms`
      
      * commit for test_onnx
      
      * remove `is_in_onnx_export` in softnms
      
      * add more comments
      
      * fix c++ lint error
      
      * add known issues doc for `cummax`
      
      * fix known issues doc
      934b549e
  25. 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
  26. 27 Apr, 2021 1 commit
  27. 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