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
