mmcv/tests
Jingwei Zhang 35affc35bf
[Feature] Support PrRoIPool operation (#2113)
* support prroipooling

* fix build bug

* fix bug of precision

* fix lint

* add ut

* add license and citation

* polish op and add type hint

* refactor ut

* add docstring for prroipool

* polish code

* format cu and cuh code

* fix format

* fix format

* fix typo

Co-authored-by: zhouzaida <zhouzaida@163.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-07-23 23:08:32 +08:00
..
data Add missing json file for testing torchvision0.13 (#2084) 2022-06-30 19:12:11 +08:00
test_cnn [Fix] num_channels can not be divisible by num_groups in test_build_norm_layer (#2085) 2022-06-30 19:28:34 +08:00
test_device [Feature] Add support for mps (#2092) 2022-07-07 16:05:49 +08:00
test_image [Feature] Add adjust_hue (#1628) 2022-05-10 17:06:45 +08:00
test_ops [Feature] Support PrRoIPool operation (#2113) 2022-07-23 23:08:32 +08:00
test_runner Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_utils Add torch_meshgrid wrapper due to PyTorch change (#2044) 2022-06-15 20:36:48 +08:00
test_video Transfer test data to aliyun (#2004) 2022-05-27 14:54:08 +08:00
test_arraymisc.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_fileclient.py [Fix] Fix the problem that the instance cannot reuse when overwriting the file client. (#1747) 2022-05-25 19:34:15 +08:00
test_fileio.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
test_load_model_zoo.py [Enhancement] Enhance get_torchvision_models (#1867) 2022-04-18 23:20:48 +08:00
test_parallel.py Fix lint (#1984) 2022-05-25 20:45:31 +08:00
test_visualization.py