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
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
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
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
51b7a6d74a
[Fix] Fix spconv unit test by releasing cache on GPU ( #1872 )
2022-04-14 22:35:29 +08:00
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
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
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
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
6289b6f922
[Fix] Fix rocm support ( #1704 )
2022-03-21 15:24:22 +08:00
68a2c0a1e2
Update unittest for mmdet3d ops ( #1811 )
2022-03-18 12:52:55 +08:00
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
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
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
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
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
6e9ce18323
Add copyright pre-commit-hook ( #1742 )
...
* first commit
* Add copyright pre-commit-hook
2022-02-24 09:24:25 +08:00
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
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
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
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
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
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
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
b0b30d0c5c
fix bug ( #1702 )
2022-02-09 14:00:12 +08:00
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
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
26c095dc11
[Fix] Fix the undefined name error ( #1695 )
2022-01-27 20:11:32 +08:00
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
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
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
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
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
227037fcd3
[Fix] Fix the unit test of correlation op ( #1659 )
2022-01-13 14:27:50 +08:00
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
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
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
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
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
304efbb650
[Feature] Add points_in_polygons CUDA op for rotated detection. ( #1600 )
2021-12-24 10:56:48 +08:00
0bcbeadb53
[Feature] Add RiRoIAlignRotated CUDA op for rotated detection. ( #1599 )
2021-12-24 00:30:34 +08:00
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
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
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
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
001f3f6610
[Feature]Add patch embed and patch merge ( #1305 )
2021-12-14 21:52:37 +08:00
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
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
88e017337a
[Fix] fix config type inconsistency ( #1575 )
...
* [Fix] fix config type inconsistency
* [Fix] Fix unit test
2021-12-14 13:17:57 +08:00
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
Jiazhen Wang
Danila Rukhovich
Charlyo
VVsssssk
Qiza-lyhm
Tan SU
Sharpiless
Tan SU
q.yao
Wenhao Wu
Alex Yang
saurbhc
Hakjin Lee
Ma Zerun
RangiLyu
Yue Zhou
Mashiro
bilibilee
imyhxy
fatih
zhanggefan
Hongbin Sun
David de la Iglesia Castro
Shilong Zhang
wuwencheng