1. 18 May, 2022 1 commit
  2. 16 Apr, 2022 2 commits
    • Zaida Zhou's avatar
      [Refactor] Move is_mlu_available to mmcv/utils/device_type.py (#1884) · 03b38a46
      Zaida Zhou authored
      * [Refactor] Move is_mlu_available to mmcv/utils/device_type.py
      
      * remove comma
      
      * fix isort
      03b38a46
    • Jiazhen Wang's avatar
      [Feature] Add several MLU ops (#1563) · 362a90f8
      Jiazhen Wang authored
      
      
      * [Feature] Add roiaware pool3d ops from mmdet3d (#1382)
      
      * add ops (roiaware pool3d) in mmdet3d
      
      * refactor code
      
      * fix typo
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      
      * [Feature] Add iou3d op from mmdet3d (#1356)
      
      * add ops (iou3d) in mmdet3d
      
      * add unit test
      
      * refactor code
      
      * refactor code
      
      * refactor code
      
      * refactor code
      
      * refactor code
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      
      * [Fix] Update test data for test_iou3d (#1427)
      
      * Update test data for test_iou3d
      
      * delete blank lines
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      
      * [Feature] Add group points ops from mmdet3d (#1415)
      
      * add op (group points) and its related ops (ball query and knn) in mmdet3d
      
      * refactor code
      
      * fix typo
      
      * refactor code
      
      * fix typo
      
      * refactor code
      
      * make input contiguous
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      
      * add mmdet3d op (#1425)
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      
      * [Feature] Loading objects from different backends and dumping objects to different backends (#1330)
      
      * [Feature] Choose storage backend by the prefix of filepath
      
      * refactor FileClient and add unittest
      
      * support loading from different backends
      
      * polish docstring
      
      * fix unittet
      
      * rename attribute str_like_obj to is_str_like_obj
      
      * add infer_client method
      
      * add check_exist method
      
      * rename var client to file_client
      
      * polish docstring
      
      * add join_paths method
      
      * remove join_paths and add _format_path
      
      * enhance unittest
      
      * refactor unittest
      
      * singleton pattern
      
      * fix test_clientio.py
      
      * deprecate CephBackend
      
      * enhance docstring
      
      * refactor unittest for petrel
      
      * refactor unittest for disk backend
      
      * update io.md
      
      * add concat_paths method
      
      * improve docstring
      
      * improve docstring
      
      * add isdir and copyfile for file backend
      
      * delete copyfile and add get_local_path
      
      * remove isdir method of petrel
      
      * fix typo
      
      * add comment and polish docstring
      
      * polish docstring
      
      * rename _path_mapping to _map_path
      
      * polish docstring and fix typo
      
      * refactor get_local_path
      
      * add list_dir_or_file for FileClient
      
      * add list_dir_or_file for PetrelBackend
      
      * fix windows ci
      
      * Add return docstring
      
      * polish docstring
      
      * fix typo
      
      * fix typo
      
      * deprecate the conversion from Path to str
      
      * add docs for loading checkpoints with FileClient
      
      * refactor map_path
      
      * add _ensure_methods to ensure methods have been implemented
      
      * fix list_dir_or_file
      
      * rename _ensure_method_implemented to has_method
      
      * Add CI for pytorch 1.10 (#1431)
      
      * [Feature] Upload checkpoints and logs to ceph (#1375)
      
      * [Feature] Choose storage backend by the prefix of filepath
      
      * refactor FileClient and add unittest
      
      * support loading from different backends
      
      * polish docstring
      
      * fix unittet
      
      * rename attribute str_like_obj to is_str_like_obj
      
      * [Docs] Upload checkpoint to petrel oss
      
      * add infer_client method
      
      * Support uploading checkpoint to petrel oss
      
      * add check_exist method
      
      * refactor CheckpointHook
      
      * support uploading logs to ceph
      
      * rename var client to file_client
      
      * polish docstring
      
      * enhance load_from_ceph
      
      * refactor load_from_ceph
      
      * refactor TextLoggerHook
      
      * change the meaning of out_dir argument
      
      * fix test_checkpoint_hook.py
      
      * add join_paths method
      
      * remove join_paths and add _format_path
      
      * enhance unittest
      
      * refactor unittest
      
      * add a unittest for EvalHook when file backend is petrel
      
      * singleton pattern
      
      * fix test_clientio.py
      
      * deprecate CephBackend
      
      * add warning in load_from_ceph
      
      * fix type of out_suffix
      
      * enhance docstring
      
      * refactor unittest for petrel
      
      * refactor unittest for disk backend
      
      * update io.md
      
      * add concat_paths method
      
      * fix CI
      
      * mock check_exist
      
      * improve docstring
      
      * improve docstring
      
      * improve docstring
      
      * improve docstring
      
      * add isdir and copyfile for file backend
      
      * delete copyfile and add get_local_path
      
      * remove isdir method of petrel
      
      * fix typo
      
      * rename check_exists to exists
      
      * refactor code and polish docstring
      
      * fix windows ci
      
      * add comment and polish docstring
      
      * polish docstring
      
      * polish docstring
      
      * rename _path_mapping to _map_path
      
      * polish docstring and fix typo
      
      * refactor get_local_path
      
      * add list_dir_or_file for FileClient
      
      * add list_dir_or_file for PetrelBackend
      
      * fix windows ci
      
      * Add return docstring
      
      * polish docstring
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * fix error when mocking PetrelBackend
      
      * deprecate the conversion from Path to str
      
      * add docs for loading checkpoints with FileClient
      
      * rename keep_log to keep_local
      
      * refactor map_path
      
      * add _ensure_methods to ensure methods have been implemented
      
      * fix list_dir_or_file
      
      * rename _ensure_method_implemented to has_method
      
      * refactor
      
      * polish information
      
      * format information
      
      * bump version to v1.3.16 (#1430)
      
      * [Fix]: Update test data of test_tin_shift (#1426)
      
      * Update test data of test_tin_shift
      
      * Delete tmp.engine
      
      * add pytest raises asserterror test
      
      * raise valueerror, update test log
      
      * add more comment
      
      * Apply suggestions from code review
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      
      * fix the wrong function reference bug in BaseTransformerLayer when batch_first is True (#1418)
      
      * [Docs] Add mmcv itself in the docs list (#1441)
      
      * Add mmcv itself in the docs list
      
      * modify link of docs
      
      * [Improve] improve checkpoint loading log (#1446)
      
      * [Feature] Support SigmoidFocalLoss with Cambricon MLU backend (#1346)
      
      * [Feature] Support SigmoidFocalLoss with Cambricon MLU backend
      
      * refactor MMCV_WITH_MLU macro define
      
      * refactor NFU_ALIGN_SIZE, PAD_DOWN and split_pipeline_num
      
      * delete extra fool proofing in cpp
      
      * [Feature] Support SigmoidFocalLossBackward with Cambricon MLU backend
      
      * fix macro definition in SigmoidFocalLoss
      
      * refactor mlu files into clang-format
      
      * refactor sigmoid focal loss test
      
      * refactor Sigmoid Focal Loss file structure.
      
      * fix python lint error
      
      * fix import torch_mlu error type
      
      * fix lint
      
      * refactor clang format style to google
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      
      * [Feature] Support RoiAlign With Cambricon MLU Backend (#1429)
      
      * [Feature] Support NMS with cambricon MLU backend (#1467)
      
      * [Feature] Support BBoxOverlaps with cambricon MLU backend (#1507)
      
      * [Refactor] Format C++ code
      
      * [Refactor] include common_mlu_helper in pytorch_mlu_helper and refactor build condition
      
      * [Improve] Improve the performance of roialign, nms and focalloss with MLU backend (#1572)
      
      * [Improve] Improve the performance of roialign with MLU backend
      
      * replace CHECK_MLU with CHECK_MLU_INPUT
      
      * [Improve] Improve the perf of nms and focallosssigmoid with MLU backend
      
      * [Improve] Improve the performance of roialign with MLU backend (#1741)
      
      * [Feature] Support tin_shift with cambricon MLU backend (#1696)
      
      * [Feature] Support tin_shift with cambricon MLU backend
      
      * [fix] Add the assertion of batch_size in tin_shift.py
      
      * [fix] fix the param check of tin_shift in cambricon code
      
      * [fix] Fix lint failure.
      
      * [fix] Fix source file lint failure.
      
      * Update mmcv/ops/tin_shift.py
      
      [Refactor] Modify the code in mmcv/ops/tin_shift.py.
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      Co-authored-by: default avatarbudefei <budefei@cambricon.com>
      Co-authored-by: default avatarbudefei <budefei@cambricom.com>
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      
      * resolve conflicts and fix lint
      
      * fix mmcv.utils.__init__
      
      * fix mmcv.utils.__init__
      
      * Fix lints and change FLAG
      
      * fix setup and refine
      
      * remove a redundant line
      
      * remove an unnecessary 'f'
      
      * fix compilation error
      Co-authored-by: default avatardingchang <hudingchang.vendor@sensetime.com>
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      Co-authored-by: default avatarq.yao <yaoqian@sensetime.com>
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      Co-authored-by: default avatarpc <luopeichao@sensetime.com>
      Co-authored-by: default avatarWenwei Zhang <40779233+ZwwWayne@users.noreply.github.com>
      Co-authored-by: default avatarq.yao <streetyao@live.com>
      Co-authored-by: default avatarTong Gao <gaotongxiao@gmail.com>
      Co-authored-by: default avatarYuxin Liu <liuyuxin@cambricon.com>
      Co-authored-by: default avatarzihanchang11 <92860914+zihanchang11@users.noreply.github.com>
      Co-authored-by: default avatarshlrao <shenglong.rao@gmail.com>
      Co-authored-by: default avatarzhouchenyang <zcy19950525@gmail.com>
      Co-authored-by: default avatarMrxiaofei <36697723+Mrxiaofei@users.noreply.github.com>
      Co-authored-by: default avatarbudefei <budefei@cambricon.com>
      Co-authored-by: default avatarbudefei <budefei@cambricom.com>
      362a90f8
  3. 24 Feb, 2022 1 commit
  4. 14 Dec, 2021 1 commit
    • Shilong Zhang's avatar
      [Feature]Support skip nms (#1552) · 43b2f098
      Shilong Zhang authored
      * skip nms
      
      * judge at beginning
      
      * add test
      
      * remove else
      
      * add more details in docstr including version not
      
      * fix unitest
      
      * fix doc
      
      * fix doc
      
      * fix typo
      
      * resove conversation
      
      * fix link
      43b2f098
  5. 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
  6. 13 May, 2021 1 commit
  7. 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
  8. 23 Aug, 2020 1 commit
  9. 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