1. 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
  2. 13 May, 2021 1 commit
  3. 26 Feb, 2021 1 commit
    • 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
  4. 23 Aug, 2020 1 commit
  5. 28 Jun, 2020 1 commit
    • zhuyuanhao's avatar
      add ext ops, support parrots (#310) · c0f5492e
      zhuyuanhao authored
      
      
      * add ext ops, support parrots
      
      * fix lint
      
      * fix lint
      
      * update op from mmdetection
      
      * support non-pytorch env
      
      * fix import bug
      
      * test not import mmcv.op
      
      * rename mmcv.op to mmcv.ops
      
      * fix compile warning
      
      * 1. fix syncbn warning in pytorch 1.5
      2. support only cpu compile
      3. add point_sample from mmdet
      
      * fix text bug
      
      * update docstrings
      
      * fix line endings
      
      * minor updates
      
      * remove non_local from ops
      
      * bug fix for nonlocal2d
      
      * rename ops_ext to _ext and _ext to _flow_warp_ext
      
      * update the doc
      
      * try clang-format github action
      
      * fix github action
      
      * add ops to api.rst
      
      * fix cpp format
      
      * fix clang format issues
      
      * remove .clang-format
      Co-authored-by: default avatarKai Chen <chenkaidev@gmail.com>
      c0f5492e