JarvisKevin
60eadb0665
[Enhancement] Lower the restrictions of _resize method in BaseMergeCell ( #1959 )
...
* Fix the bug met in using nasfpn
Fix the bug met in using nasfpn which is mentioned at https://github.com/open-mmlab/mmdetection/issues/5987 .
Avoid the strong restrictions of _resize function in BaseMergeCell:
1. When Downsampling the feature map, the feature map's shape must be divisible by the target size. We pad zero around feature map before max_pool2d opt to make it always divisible. (line 102 ~ 107)
2. Considering the different downsampling scale of H and W, shape[-2] and shape[-1] are involed in the definition of kernel_size. (line 110)
* Update merge_cells.py
check flake8 & isort
* Update merge_cells.py
* Update merge_cells.py
yapf
* Update mmcv/ops/merge_cells.py
X_pad rename to padding_x
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
* Update merge_cells.py
format the code style after renameing the X_pad to padding_x
* Update test_merge_cells.py
Mainly test the downsampling resize in BaseMergeCell. The smaller target size is set to (14, 7), the classical feature map's size in the last few stages of the backbone, which will product different downsampling scales in different dims.
* Update test_merge_cells.py
add "# Copyright (c) OpenMMLab. All rights reserved."
* Update merge_cells.py
format the variable name
* Update test_merge_cells.py
Testing divisible and indivisible situations simultaneously
* Update mmcv/ops/merge_cells.py
fix the bug when h is indivisible and w is divisible, the pad_w will be padded unreasonable.
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/merge_cells.py
fix the bug when w is indivisible and h is divisible, the pad_h will be padded unreasonable.
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix undefined error
* Update merge_cells.py
make pad_h, pad_w more readable
* Update test_merge_cells.py
use @pytest.mark.parametrize instead of 'for' methor
* Update merge_cells.py
* Update test_merge_cells.py
isort
* Update merge_cells.py
isort
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-20 15:13:13 +08:00
Zaida Zhou
45fa3e44a2
Add pyupgrade pre-commit hook ( #1937 )
...
* add pyupgrade
* add options for pyupgrade
* minor refinement
2022-05-18 11:47:14 +08:00
Jiazhen Wang
105fe55002
[Feature] Add adjust_hue ( #1628 )
...
* [Impl] impl adjust hue
* Update docstring
* Fix docstring
* add copyright
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-10 17:06:45 +08:00
jmercat
1dbb5d360d
[Fix] Fix dump method of Config ( #1837 )
...
* config dump fix
dump should not depend on how config was loaded only where it is dumped
changed checks for type to the given filename instead of the source filename
* defined test_fromdict and fixed dump function
it might be more convoluted than it needs to be but the formatting depends on the extension of the argument or the extension of the self.filename
* config dump defaults to returning pretty_text, test_fromdict renamed to test_dump_from_dict
* some reformatting in docstrings
* refine unittest
* fix unit test as comment
* Refine docstring
* minor refinement
* import mmcv
* fix lint
Co-authored-by: HAOCHENYE <21724054@zju.edu.cn>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-10 15:41:51 +08:00
pc
e70a318ae9
[Fix] Fix parrots unit tests ( #1913 )
2022-05-09 16:59:32 +08:00
Wencheng Wu
a3a078eefb
[Enhancement] Support registering function ( #1858 )
...
* [Enhance] Support register function.
* fix unittest error
* add docs and unittest of register function
* modify the docs
* fix version to 1.5.1
* Update docs/zh_cn/understand_mmcv/registry.md
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update docs/en/understand_mmcv/registry.md
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* refine the docs
* modify module_class to module
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-02 18:31:41 +08:00
WilliamKyle
b30755eef2
[Feature] Add rotated_feature_align cpu & onnxruntime implementation ( #1878 )
...
* add rotated_feature_align cpu implementation
* add rotated_feature_align onnxruntime implementation
* Update code for advices from grimoire
Remove useless comment from mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp
Replace ambiguous function name atomicAdd in mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp
Simplify unit test with parameter in tests/test_ops/test_rotated_feature_align.py
Use fma in interpolate in mmcv/ops/csrc/onnxruntime/cpu/rotated_feature_align.cpp mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp
* Inline the function to reduce the overhead of the function call
Use fma in interpolate
2022-04-29 10:02:33 +08:00
Jiazhen Wang
c324b1fcfc
[Fix] Ensure the type of filename parameter in Config is str ( #1725 )
...
* ensure type of filename is str
* check filename for func: fromfile
* add ut for fromfile
2022-04-27 19:50:04 +08:00
q.yao
de0c1039f7
Fix bbox unit test ( #1914 )
2022-04-26 16:49:28 +08:00
q.yao
0e59036211
[Enhancement] Optimize bbox overlap ( #1718 )
...
* add half support
* add cpu implementation
* fix bugs, load with inline asm
* better vector load
* add comments
2022-04-26 00:44:01 +08:00
Artem
94c071b310
[Feature] Add ClearMLLoggerHook ( #1906 )
...
* added clearml logger support
* review fixes
* review fixes
2022-04-24 13:20:09 +08:00
Mashiro
a80df6874d
[Enhancement] Enhance get_torchvision_models ( #1867 )
...
* enhance get_torchvision_models
* simplify logic
* Dump ckpt in torchvision lower than 0.13.0 to a json file
* add json
* refactor load urls logic
* fix unit test
* change url key to lower letters
* check torchvision version rather than check torch version in unittest
* Fix CI and refine test logic of torchvision version
* add comment
* support compare pre-release version
* support loaad modeel like torchvision
* refine comment.
* fix unit test and comment
* fxi unit test bug
* support get model by lower weights
2022-04-18 23:20:48 +08:00
Vince Jankovics
b80447707c
[Enhancement] Support None in DictAction ( #1834 )
...
* None parsing fixed for config
* Formatting
* Lower case none fix
2022-04-18 14:44:32 +08:00
Hu Di
5221a3883c
[Feature] Support MMCV on IPU ( #1882 )
...
* implement runner on IPU
* adjust import
* adjust import
* add ignore for ipu on without ipu
* remove compilation cache
* remove ipu from mmcv/runner.__all__
* adjust IS_IPU and IS_MLU
* adjust by isort
* add ipuHardwareIsAvailable
* remove engine_cache
* code review 9
2022-04-18 00:02:49 +08:00
Zaida Zhou
03b38a46ad
[Refactor] Move is_mlu_available to mmcv/utils/device_type.py ( #1884 )
...
* [Refactor] Move is_mlu_available to mmcv/utils/device_type.py
* remove comma
* fix isort
2022-04-17 00:15:35 +08:00
Jiazhen Wang
362a90f8bf
[Feature] Add several MLU ops ( #1563 )
...
* [Feature] Add roiaware pool3d ops from mmdet3d (#1382 )
* add ops (roiaware pool3d) in mmdet3d
* refactor code
* fix typo
Co-authored-by: zhouzaida <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: zhouzaida <zhouzaida@163.com>
* [Fix] Update test data for test_iou3d (#1427 )
* Update test data for test_iou3d
* delete blank lines
Co-authored-by: Zaida 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: zhouzaida <zhouzaida@163.com>
* add mmdet3d op (#1425 )
Co-authored-by: zhouzaida <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: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida 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: zhouzaida <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: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: budefei <budefei@cambricon.com>
Co-authored-by: budefei <budefei@cambricom.com>
Co-authored-by: Zaida 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: dingchang <hudingchang.vendor@sensetime.com>
Co-authored-by: zhouzaida <zhouzaida@163.com>
Co-authored-by: q.yao <yaoqian@sensetime.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: pc <luopeichao@sensetime.com>
Co-authored-by: Wenwei Zhang <40779233+ZwwWayne@users.noreply.github.com>
Co-authored-by: q.yao <streetyao@live.com>
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
Co-authored-by: Yuxin Liu <liuyuxin@cambricon.com>
Co-authored-by: zihanchang11 <92860914+zihanchang11@users.noreply.github.com>
Co-authored-by: shlrao <shenglong.rao@gmail.com>
Co-authored-by: zhouchenyang <zcy19950525@gmail.com>
Co-authored-by: Mrxiaofei <36697723+Mrxiaofei@users.noreply.github.com>
Co-authored-by: budefei <budefei@cambricon.com>
Co-authored-by: budefei <budefei@cambricom.com>
2022-04-16 15:45:00 +08:00
Danila Rukhovich
aee596d523
[Enhance] Differentiable rotated IoU ( #1854 )
...
* diff_iou_rotated is working
* add test; fix lint
* fix lint for test
* disable cpu build
* refactor files structure
* fix comments
* remove extra .repeat()
* add comment
* fix j-1 bug; update doc
* fix clang lint
* update docstrings
* fix comments
* fix comments
2022-04-15 15:40:07 +08:00
Charlyo
7982dd1a06
Make sure padding size is not negative ( #1792 )
...
* Make sure padding size is not negative
If Resize maintains aspect ratio, could be the case that image shape is bigger than resize value.
* Test pad shape smaller than image shape
* Fix lint
* Remove whitespaces
2022-04-15 14:05:32 +08:00
VVsssssk
51b7a6d74a
[Fix] Fix spconv unit test by releasing cache on GPU ( #1872 )
2022-04-14 22:35:29 +08:00
Qiza-lyhm
4826a9b7e4
[Feature] Support PyTorch backend on MLU ( #1770 )
...
* feat(MLU): Support PyTorch backend on MLU
* MMCV support PyTorch backend on MLU
* Add MLUDataParallel and MLUDistributedDataParallel
* Add MLU operator support
* [Fix]: Fix PR comments and add IS_MLU to get device available check
* [Fix]: fix PR comments of dist_utils.py
* [Doc] Rewrite annotations of functions.
* [Docs] Rewrite annotation in distributed.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* [Docs] Fix lint
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-04-14 21:48:18 +08:00
Tan SU
c33f248987
[Enhancement] Fix collect_env on Windows ( #1789 )
...
* [fix] fix collect_env() on Windows
* fix(utils): add back 'Cuda compilation tools' to nvcc info
* feat(utils): find default ccompiler
* fix(utils, test): parse cl.exe version instead of path
* fix: code style
* fix(tools): use distutils
* [fix] collect_env() returns GCC or MSVC
* [fix] unitest for collect_env()
* add comments
* test: add GCC to expected_keys
* fix: code style
* Update mmcv/utils/env.py
2022-04-05 16:31:16 +08:00
Sharpiless
969e2af866
[Feature] Support for linearly learning rate decay ( #1627 )
...
* update support for linearly learning rate decay
* Fix LinearAnnealingLrUpdaterHook, update LinearAnnealingMomentumUpdaterHook, add unit test
add docstring
add docstring
update linear lr momentum schedule test
fix ci
Fix CI
* add lr and momentum hook to runner and hooks package
add lr and momentum hook to runner and hooks package
* replace multi_optimziers with multi_optimizers
Co-authored-by: HAOCHENYE <21724054@zju.edu.cn>
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
2022-03-31 09:39:40 +08:00
Tan SU
ac52bb3795
[fix] config ignore imported modules and functions ( #1802 )
...
* [fix] config ignore modules and functions
* add unitest
2022-03-22 23:04:55 +08:00
q.yao
6289b6f922
[Fix] Fix rocm support ( #1704 )
2022-03-21 15:24:22 +08:00
Wenhao Wu
68a2c0a1e2
Update unittest for mmdet3d ops ( #1811 )
2022-03-18 12:52:55 +08:00
Alex Yang
1cd864d634
[Feature] Support shallow copy for Config ( #1796 )
...
* Add .owners.yml to mark daily issue shift
* Update .owners.yml
fix end of file
* Add __copy__() to Config
* fix format in config.py
* fix format in test_copy
* fix format in config.py
* Update config.py
* Update tests/test_utils/test_config.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-03-15 14:25:01 +08:00
Wenhao Wu
b5d550f090
[Feature] Add nondeterministic voxelization op from mmdet3d ( #1783 )
...
* add nondeterministic voxelization op
* fix lint
* fix lint
* resolve comments
* fix lint
2022-03-15 14:21:34 +08:00
Jiazhen Wang
33e14deaea
[Enhancement] Add PyTorch1.11.0 CI ( #1800 )
...
* add pytorch1.11 ci
* fix test parallel
* fix test parallel
2022-03-15 12:38:30 +08:00
Wenhao Wu
09b64a60b0
[Fix] Fix bugs in DynamicScatter op ( #1748 )
...
* Fix bugs in DynamicScatter op
* recover unittest
* add a comment as a reminder
* compatible to torch with lower version
2022-03-07 21:02:38 +08:00
saurbhc
34f227efbd
[Feature] Add SegmindLoggerHook ( #1650 )
...
* add SegmindLoggerHook
* update linting for segmind.py
* ran pre-commit
* add test_segmind_hook
- add SegmindLoggerHook import in:
mmcv/runner/__init__.py
mmcv/runner/hooks/__init__.py
mmcv/runner/hooks/logger/__init__.py
* update test_segmind_hook
- Add Docstring to SegmindLoggerHook
- Use get_loggable_tags(...)
* update test_hooks.py & segmind.py
- mmcv/runner/hooks/logger/segmind.py
moved docs from __init__ to class ...
update ImportError line-indentation
remove unwanted method
- tests/test_runner/test_hooks.py
update assert_called_with only on hook.segmind_mlflow_log
* Update tests/test_runner/test_hooks.py
disable yapf on test_hooks.py imports
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/runner/hooks/logger/segmind.py
Update SegmindLoggerHook docstring
Co-authored-by: Jiazhen Wang <47851024+teamwong111@users.noreply.github.com>
* update before_run method in segmind.py
removed un-used statements
* updated imports in SegmindLoggerHook
* update SegmindLoggerHook
- code cleanup
* update SegmindLoggerHook
- add interval parameter in __init__ method
* update SegmindLoggerHook
- more arguments passes to __init__ method
- interval
- ignore_last
- reset_flag
- by_epoch
* Update mmcv/runner/hooks/logger/segmind.py
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
* Update mmcv/runner/hooks/logger/segmind.py
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Jiazhen Wang <47851024+teamwong111@users.noreply.github.com>
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
2022-03-03 10:01:46 +08:00
Zaida Zhou
6e9ce18323
Add copyright pre-commit-hook ( #1742 )
...
* first commit
* Add copyright pre-commit-hook
2022-02-24 09:24:25 +08:00
Zaida Zhou
e7f049afa1
Add mmrotate link in README.md ( #1733 )
...
* Add mmrotate link in README.md
* Add mmrotate link in docs
* fix isort
2022-02-20 11:42:33 +08:00
Hakjin Lee
b83bdb0cbe
[Refactor] Refactor the interface for RoIAlignRotated ( #1662 )
...
* fix interface for RoIAlignRotated
* Add a unit test for RoIAlignRotated
* Make a unit test for RoIAlignRotated concise
* fix interface for RoIAlignRotated
* Refactor ext_module.nms_rotated
* Lint cpp files
2022-02-18 16:59:33 +08:00
Wenhao Wu
c1de4c9bbf
[Feature] Add spconv ops from mmdet3d ( #1581 )
...
* add ops (spconv) of mmdet3d
* fix typo
* refactor code
* resolve comments in #1452
* fix compile error
* fix bugs
* fix bug
* transform from 'types.h' to 'extension.h'
* fix bug
* transform from 'types.h' to 'extension.h' in parrots
* add extension.h in pybind.cpp
* add unittest
* Recover code
* (1) Remove prettyprint.h
(2) Switch `T` to `scalar_t`
(3) Remove useless lines
(4) Refine example in docstring of sparse_modules.py
* (1) rename from `cu.h` to `cuh`
(2) remove useless files
(3) move cpu files to `pytorch/cpu`
* reorganize files
* Add docstring for sparse_functional.py
* use dispatcher
* remove template
* use dispatch in cuda ops
* resolve Segmentation fault
* remove useless files
* fix lint
* fix lint
* fix lint
* fix unittest in test_build_layers.py
* add tensorview into include_dirs when compiling
* recover all deleted files
* fix lint and comments
* recover setup.py
* replace tv::GPU as tv::TorchGPU & support device guard
* fix lint
Co-authored-by: hdc <hudingchang.vendor@sensetime.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
2022-02-18 15:53:10 +08:00
Ma Zerun
33c83b5afa
Support deepcopy for Config ( #1658 )
...
* Support deepcopy for Config
* Iterate the `__dict__` of Config directly.
* Use __new__ to avoid unnecessary initialization.
* Improve according to comments
2022-02-16 22:51:08 +08:00
Hakjin Lee
b10eaedd38
[Fix] batched nms for nms_rotated ( #1688 )
...
* Fix batched_nms to support nms_rotated
* Update test code for batched_nms with nms_rotated
* Edit indexing the score in nms
* Use -1 indexing for batched_nms
2022-02-12 14:45:39 +08:00
RangiLyu
62c1b7f68b
[Enhance] Use PyTorch HSwish implementation. ( #1709 )
...
* [Enhance] Use PyTorch HSwish implementation.
* fix conv test
* upgrade version
* add version comments
2022-02-12 14:34:35 +08:00
Zaida Zhou
dd82ab1497
Upgrade the version of isort ( #1705 )
...
* Upgrade the version of isort
* sort the order of importing modules
2022-02-09 21:29:21 +08:00
Yue Zhou
b0b30d0c5c
fix bug ( #1702 )
2022-02-09 14:00:12 +08:00
Hakjin Lee
7df4ebedb3
[Fix] Fix the mismatch torch version of MlflowLoggerHook( #1680 )
...
* [Fix] mlflow logger error
* [Update] mlflow hook test arguments
* Replace torch.__version__ with mmcv.TORCH_VERSION
* update test code
2022-02-07 23:14:46 +08:00
Mashiro
26aba2f533
[Fix] Fix an incorrect assignment in the BaseTransformerLayer ( #1670 )
...
* fix BaseTransformerLayer
* Add BaseTransformerLayer unit test without ffn_cfg embed_dims
* Optimize code logic
2022-01-28 23:43:53 +08:00
Hakjin Lee
26c095dc11
[Fix] Fix the undefined name error ( #1695 )
2022-01-27 20:11:32 +08:00
Zaida Zhou
9dedc44555
[Fix] Fix the cast_tensor_type ( #1639 )
...
* [Fix] Fix the cast_tensor_type when the type of inputs is not the same as src_type
* Add note
* improve comments
2022-01-24 21:21:37 +08:00
Mashiro
580e374eae
Add decay to CyclicLrUpdater ( #1655 )
...
* Add decay to cyclic LR
* Simplify tests
* Fix error in tests
* fix cyclic lr
* add weight decay CyclicLrUpdate test
* Update mmcv/runner/hooks/lr_updater.py
update docstring
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* update CyclicLrUpdater unit test
* add comments to CyclicLrUpdaterHook
* fix CyclicLrUpdater. Support peak_lr decay when target[0] < 1
* add momentum decay to CyclicMomentumUpdater
* update momentum unit test
* Fix CyclicMomentum comment
Fix CyclicMomentum comment
Co-authored-by: Dmytro Panchenko <panchenkodmytry@gmail.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-01-24 19:34:25 +08:00
bilibilee
0448fcf906
fix the scatter when input is cpu ( #1621 )
...
* fix the scatter when input is cpu
* Update _functions.py
Add spaces to comply with the code specification
* add unittests
* add a blank line
* fix unittet
Co-authored-by: zhouzaida <zhouzaida@163.com>
2022-01-24 19:17:10 +08:00
Jiazhen Wang
b8d78336a7
[Fix] Align HSigmoid with pytorch official implementation( #1622 )
...
* [Fix] align hsigmoid with pytorch official
* [Fix] add warnings for Hsigmoid
* [Fix] fix format
* [Fix] add unittest
* [Fix] fix docstring
2022-01-20 14:25:10 +08:00
imyhxy
37121bf4f4
[Fix] Fix the bug of momentum updater with warmup policy ( #1626 )
...
* fix momentum updater typo bug
* Add unit test for 'constant', 'linear' and 'exp' momentum warmup policy
2022-01-18 17:33:07 +08:00
Zaida Zhou
227037fcd3
[Fix] Fix the unit test of correlation op ( #1659 )
2022-01-13 14:27:50 +08:00
Yue Zhou
594ff3c0d3
[Feature] Add convex IoU CUDA op for rotated detection. ( #1610 )
...
* add convex iou
* fix lint
* add convex_iou
* fix convex_iou
* add convex_giou
* fix bug
* fix lint
* fix bug
* Update
* update
* add kernel loop
* fix bug.
* fix polygen typo
* simplify reverse
* Update convex_iou_cuda_kernel.cuh
* Update mmcv/ops/convex_iou.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/convex_iou.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* add AT_DISPATCH_FLOATING_TYPES_AND_HALF
* fix lint
* fix lint
* Resolving conflicts
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-01-10 20:43:12 +08:00
Yue Zhou
9acc892a44
[Feature] Add active rotated filter op for rotated detection. ( #1598 )
...
* add active_rotated_filter
* fix lint
* fix lint
* renaming nRotation and nOrientation
* Update active_rotated_filter_cuda_kernel.cuh
* Update active_rotated_filter_cuda.cu
* fix bug
* fix lint
* Update test_active_rotated_filter.py
* fix lint
* Update active_rotated_filter_cuda_kernel.cuh
* renaming
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix lint
* Resolving conflicts
* fix lint.
* Update __init__.py
* Update mmcv/ops/csrc/pytorch/cuda/active_rotated_filter_cuda.cu
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update active_rotated_filter.cpp
* fix lint
* Update mmcv/ops/csrc/pytorch/cpu/active_rotated_filter.cpp
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/csrc/pytorch/cpu/active_rotated_filter.cpp
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update active_rotated_filter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-01-10 17:35:20 +08:00
fatih
40518322b6
add artifact logging to wandb hook ( #1616 )
...
* add artifact logging to wandb hook
* upload artifacts wiwth only specified suffix
* update docstring
* Update mmcv/runner/hooks/logger/wandb.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix linting
* add tests for wandb artifact logging
* remove redundant lines
* fix wandb tests
* init `WandbLoggerHook` with `log_artifact=True` in tests
* remove redundant lines from wandb tests
* add docstring for `with_step`
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-01-10 16:52:38 +08:00
Yue Zhou
51b40c332a
[Feature] Add min_area_polygons CUDA op for rotated detection. ( #1611 )
...
* init
* Update pybind.cpp
* Update min_area_polygons_cuda.cuh
* Update cudabind.cpp
* fix bug
* Create test_min_area_polygons.py
* add test
* update
* Update min_area_polygons_cuda.cuh
* fix bugs.
* Update min_area_polygons_cuda.cuh
* Update min_area_polygons.py
* Update min_area_polygons_cuda.cuh
* merge these 4 nested loops
* add AT_DISPATCH_FLOATING_TYPES_AND_HALF
* fix lint
* Resolving conflicts
2022-01-10 11:00:50 +08:00
zhanggefan
b6167d5987
[Enhancement] Add clockwise argument to ops box_iou_rotated and nms_rotated ( #1592 )
...
* add clockwise arguments to ops box_iou_rotated and nms_rotated
* refactor docs
* change code that may incur stopped gradient.
* refactor docs
* Update mmcv/ops/nms.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/box_iou_rotated.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-01-10 10:57:11 +08:00
zhouyue
304efbb650
[Feature] Add points_in_polygons CUDA op for rotated detection. ( #1600 )
2021-12-24 10:56:48 +08:00
zhouyue
0bcbeadb53
[Feature] Add RiRoIAlignRotated CUDA op for rotated detection. ( #1599 )
2021-12-24 00:30:34 +08:00
zhouyue
2475dc3452
[Feature] Add feature refine CUDA op for rotated detection. ( #1603 )
...
* re PR
* replace all the feature_refine with rotated_feature_align
* replace all the FR with RotatedFeatureAlign
* Update mmcv/ops/rotated_feature_align.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/rotated_feature_align.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix lint
* Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/csrc/pytorch/cuda/cudabind.cpp
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix lint
* Update cudabind.cpp
* Update cudabind.cpp
* fix bug in test. & add backward test
* fix lint
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-12-23 17:38:58 +08:00
Hongbin Sun
690a77fa6f
[Feature]: Support tensor2grayimgs ( #1595 )
...
* support tensor2grayimgs
* give default mean and std according to the input channel
* update docstring
* update
* fix bug
2021-12-22 19:49:50 +08:00
David de la Iglesia Castro
ac92a1116f
`DvcliveLoggerHook` updates to work with `DVC` ( #1208 )
...
* Updates to work with DVC
* Update docstrings
* Updated test
* Updated DVCLiveLoggerHook
* Fix name
* Added missing next_step call
* Fix expected call
* Implicit next_step
* Suggestions from review
* Update test_hooks.py
* Updated to last dvclive version
* Cleaned docstring
* Update mmcv/runner/hooks/logger/dvclive.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update dvclive.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-12-22 19:35:06 +08:00
Jiazhen Wang
fb486b96fd
[Fix] Fix some warnings in unittest ( #1522 )
...
* [Fix] fix some warnings in unittest
* [Impl] standardize some warnings
* [Fix] fix warning type in test_deprecation
* [Fix] fix warning type
* [Fix] continue fixing
* [Fix] fix some details
* [Fix] fix docstring
* [Fix] del useless statement
* [Fix] keep compatibility for torch < 1.5.0
2021-12-22 10:57:10 +08:00
Shilong Zhang
001f3f6610
[Feature]Add patch embed and patch merge ( #1305 )
2021-12-14 21:52:37 +08:00
wuwencheng
66bff139e4
[Feature] Add multi file backends to imread/imwrite. ( #1527 )
...
* Add file client to image io
* Fix petrel_client imwrite error
* Add examples to the docstring and delete the file check of imread
* modify docstring v1.3.19->v1.4.1
* Deprecate auto_mkdir parameter and complete test_io.py
* Fix error caused by deleting the mock package in test_io.py
* Add annotation to imencode
* modify imread input assert and delete the judgement of file client 'put' method
* Delete try except in imwrite.
* Add a error file extension unit test.
2021-12-14 14:37:21 +08:00
Shilong Zhang
43b2f0981c
[Feature]Support skip nms ( #1552 )
...
* 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
2021-12-14 13:18:30 +08:00
Jiazhen Wang
88e017337a
[Fix] fix config type inconsistency ( #1575 )
...
* [Fix] fix config type inconsistency
* [Fix] Fix unit test
2021-12-14 13:17:57 +08:00
Jiazhen Wang
250fadc210
[Fix] Fix checkpoint local files detect ( #1549 )
...
* [fix] fix checkpoint local files detect
* [Fix] add support for path like '~/xx/file'
* [Fix] fix some details
* [Fix] fix unittest
2021-12-14 13:17:29 +08:00
Zaida Zhou
84c7dc3436
[Fix] Fix the path of ceph containing cluster name ( #1577 )
...
* [Fix] Fix the path of ceph containing cluster name
* fix corner cases
* fix format
* add note
2021-12-14 10:41:27 +08:00
Shilong Zhang
22e73d6986
[Feature] Detect anomalous parameters ( #1547 )
...
* detect detect_anomalous_params
* fix default value
* merge two case
* fix none case
* add unitest
* fix typo
* change level to error
* fix type
* add more details in docstr
2021-12-14 10:28:36 +08:00
Luting Wang
56e9f73c58
[Feature] Add ModuleDict inherited from BaseModule and ModuleDict ( #1542 )
...
* improve: add module dict
* Update __init__.py
* docs: add module dict zh
* docs: add module dict
* Update test_basemodule.py
* Update __init__.py
* lint
* lint
* docs
* docs
* Update base_module.py
* lint
2021-12-06 11:56:12 +08:00
Zaida Zhou
990d8b6a83
[Fix] Add load_url to handle incompatibility of PyTorch versions ( #1377 )
...
* [Fix] Fix torch.load error
* [Fix] Fix torch.load error
* rename _save to _save_ckpt
* add load_url to handle imcompatibility of PyTorch versions
* add unittest for load_url
* fix typo
* print a friendly information when error occurred
2021-11-19 17:34:18 +08:00
WangJiaZhen
e3c63f34bc
[Fix] Fix deform conv by adding an extra argument im2col_step ( #1459 )
...
* [Fix] fix deform conv by add argument
* [Fix] replace useless func with np.repeat and add necessary comment
* [Fix] reduce batch_size and remove useless lines and modify some var name
* [Fix] change position of comments
* [Fix] add im2col_step in the docstring and add two test cases
* [Fix] fix docstring and add comments for test cases
* [Fix] fix docstring
* [Fix] add note, fix issue link and other details
* [Fix] fix docstring details
* [Fix] fix links in docstring
* [Fix] fix docstring details
2021-11-10 19:05:20 +08:00
zhouzaida
0633f91139
print a warning information when eval_res is an empty dict
2021-11-10 15:44:48 +08:00
zhouzaida
f2d11076eb
[Fix] Raise AssertError when eval_res is a null dict
2021-11-10 15:44:48 +08:00
Tong Gao
c522b47e6a
fix the wrong function reference bug in BaseTransformerLayer when batch_first is True ( #1418 )
2021-11-02 20:57:50 +08:00
q.yao
426e229d88
[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: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-11-02 20:50:50 +08:00
Zaida Zhou
32e09f4933
[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
2021-10-24 14:26:52 +08:00
Zaida Zhou
5b5b47fc87
[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
2021-10-23 20:51:44 +08:00
dingchang
75cae78c55
[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: zhouzaida <zhouzaida@163.com>
2021-10-23 14:01:31 +08:00
q.yao
6935a818ec
[Fix] Update test data for test_iou3d ( #1427 )
...
* Update test data for test_iou3d
* delete blank lines
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-10-22 21:18:19 +08:00
dingchang
2157b708fb
[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: zhouzaida <zhouzaida@163.com>
2021-10-22 10:46:55 +08:00
dingchang
4c8bfb48ef
[Feature] Add roiaware pool3d ops from mmdet3d ( #1382 )
...
* add ops (roiaware pool3d) in mmdet3d
* refactor code
* fix typo
Co-authored-by: zhouzaida <zhouzaida@163.com>
2021-10-21 20:00:04 +08:00
dingchang
e3e1dba2c8
[Feature] Add voxel ops from mmdet3d ( #1381 )
...
* add ops (voxel) in mmdet3d
* add ops (voxel) in mmdet3d
* add ops (voxel) in mmdet3d
* refactor code
* update test
* update test
* update test
* refactor code
* refactor code
* refactor code
* fix typo
* fix typo
2021-10-21 11:08:11 +08:00
Ezra-Yu
e8489a7b4b
Add case case_sensitive in scandir ( #1389 )
...
* add case_insensitive
* rename v
* case_insensitive to case_sensitive
* Update docstring
2021-10-20 14:34:34 +08:00
q.yao
91b8478c84
Reduce ms_deformable_attn test memory usage ( #1407 )
2021-10-15 19:49:39 +08:00
dingchang
5c25ae1a19
[Feature] Add roipoint_pool3d op from mmdet3d ( #1358 )
...
* add ops (roipoint_pool3d) in mmdet3d
* refactor code
* fix typo
* add unit test
* refactor code
* refactor code
* refactor code
* fix typo
2021-10-15 17:15:53 +08:00
dingchang
e99bfce608
[Feature] Add assign_score_withk op for paconv from mmdet3d ( #1357 )
...
* add ops (assign_score_withk) in mmdet3d
* add ops (assign_score_withk) in mmdet3d
* refactor code
* fix typo
* refactor code
* fix typo
* polish docstring
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-10-15 13:18:31 +08:00
dingchang
be5841e45d
[Feature] Add interpolate ops from mmdet3d ( #1355 )
...
* add ops (interpolate) in mmdet3d
* refactor code
* fix typo
* fix typo
* fix typo
* refactor code
2021-10-14 20:51:02 +08:00
Zaida Zhou
97e5bada4c
continue PR #1223 ( #1404 )
...
* fix MultiScaleDeformableAttention inference issue on cpu model
* fix lint
* add unintest
* remove some code
* Update tests/test_ops/test_ms_deformable_attn.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix device
* remove device
* add more device
* refactor unittest
Co-authored-by: zhicheng huang <zhichenghzc@gmail.com>
Co-authored-by: zhangshilong <2392587229zsl@gmail.com>
Co-authored-by: Shilong Zhang <61961338+jshilong@users.noreply.github.com>
2021-10-14 20:50:38 +08:00
dingchang
1cd01db935
[Feature] Add gather points op from mmdet3d ( #1338 )
...
* add ops (gather points) in mmdet3d
* add ops (gather points) in mmdet3d
* refactor code
* refactor code
* fix typo
2021-10-14 16:58:12 +08:00
dingchang
42c9e71120
[Feature] Add knn op from mmdet3d ( #1354 )
...
* add ops (knn) in mmdet3d
* refactor code
* refactor code
* fix typo
* fix typo
* fix typo
* refactor code
* refactor code
* spell typo
* fix spell typo
2021-10-14 13:33:13 +08:00
Shilong Zhang
fa22d9db4e
Fix potiential bug in `deprecated_api_warning` ( #1395 )
...
* assert both dep and new args exist at same time
* add unintest
* Update tests/test_utils/test_misc.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_misc.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* polish warning
* polish warning
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-10-13 20:53:11 +08:00
dingchang
63a6cbe913
[Feature] Add fps op from mmdet3d ( #1337 )
...
* add ops (furthest point sample) in mmdet3d
* refactor code
* refactor code
* fix typo
* fix typo
* fix typo
* refactor code
* fix typo
* define DIVUP in common_cuda_helper.hpp
2021-10-13 20:39:31 +08:00
Eugene Liu
599163e645
Add Deformable Conv CustomOp for onnxruntime ( #1343 )
...
* add onnx dcn
* replace gemm with torch C++ api
* small update
* fix cpp clang format
* pefer generic GEMM than torch ATen library
* addressing comments
* add ops path check
2021-10-13 20:28:24 +08:00
Zaida Zhou
f22c9eb4a4
Add code spell hook to pre-commit and fix typos ( #1384 )
...
* Add code-spell hook to pre-commit
* Add code-spell hook to pre-commit and fix typos
* fix setup.cfg
2021-10-04 20:13:54 +08:00
Miao Zheng
745aa7373e
[Fix] Revise unit test of correlation ( #1368 )
...
* [Fix] Revise unit test of correlation
* rename
* lint
* lint
* lint
* lint
2021-09-25 21:13:34 +08:00
Miao Zheng
324e9f18ff
[Features] Load optical flow data from bytes ( #1362 )
...
* [Features] Load optical flow data from bytes
* docstring
* revise base on comments
* compression test data
* compression test data
* docstring
* minors
2021-09-23 21:25:59 +08:00
Miao Zheng
b92ea0b5df
[Feature] Add Correlation CUDA op ( #1361 )
2021-09-23 21:08:41 +08:00
dingchang
f3dfc4135b
[Feature] Add ballquery op from mmdet3d ( #1332 )
2021-09-23 21:08:00 +08:00
Wenwei Zhang
4e101e0b75
[Feature]: Support empty tensor in MMSyncBN ( #1205 )
...
* [Feature]: support empty tensor in MMSyncBN
* refine code
* resolve comments
* clean unnecessary comments
* fix inaccurate statistics when empty tensor
* resolve comments and add docstrings
* update unit tests
* rephrase, ready for merge
2021-09-23 17:10:38 +08:00
Leojc
2a3d2d4a35
Reimplement cc_attention using pure pytorch ( #1201 )
...
* Reimplement cc_attention using pure pytorch
* fix: avoid BC-Breaking
* delete cc_attention related cpp and cuda files
* delete cc_attention related lines in pybind.cpp
* make out Tensor contiguous.
* remove unneeded lines.
* Update mmcv/ops/cc_attention.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update TestCrissCrossAttention
* passing pre-commit
* Update docstring of CrissCrossAttention
* Update docstring of CrissCrossAttention
* Update mmcv/ops/cc_attention.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* [docs]polish the docstring
* [Docs] Polish the docstring
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-09-09 21:53:45 +08:00
Tong Gao
642d281823
[Feature] Add revert_sync_batchnorm ( #1253 )
...
* [Feature] Add revert_sync_batchnorm
* support mmsyncbn (to be tested)
* Test passed
* Update docstring, rename the test file
* remove test_sync_bn
* add comment
* add mmcv.ops check
* Add a comment
* Add notes and relax test req
Co-authored-by: gaotongxiao <gaotongxiao@gmail.con>
2021-09-08 10:59:39 +08:00
young
99088c81a8
[Fix] Restrict the warning message ( #1267 )
...
* restrict the warning message
* and an important keyword in warning description
* a more elegant way of condition
* link format code too long
* fix the stupid spelling mistake
* Use issubclass to restrict warning message.
* maybe this version is more elegant.
* conv + bias + norm warning pytest
* 'created' a warning, hahaha
* isort and yapf format revision
* isort and yapf format revision
* flake8 fail issue
* I have to right this way in order to solve the conflicts between yapf and flake8, sigh...
* fixed test bug
* Add ruby windows installer source.
* Simplified the code and remove ruby source from CONTRIBUTING.md
* use _BatchNorm to simplify the code
* bug fix and add instanceNorm case into warning
* change the warning message to make it more clear
* fix unit test
2021-09-07 19:45:14 +08:00