Commit Graph

547 Commits (eaf25af6c4ca18d7f3c74c3849989e517168d9a8)
 

Author SHA1 Message Date
Jintao Lin eaf25af6c4
use CUDA_HOME from parrots_wrapper (#678) 2020-12-01 10:37:12 +08:00
WRH d2ca6cebb3
[DOC] build on windows (#663)
* previous doc

* init version of windows installation

* refine doc

* merge build instructions

* fix styles

* add some detail

* change an item to note

Co-authored-by: Kai Chen <chenkaidev@gmail.com>
2020-12-01 10:36:20 +08:00
Y. Xiong 91a7fee03a
bump version to 1.2.1 (#672) 2020-11-26 15:58:37 +08:00
Wang Xinjiang 0d562118b5
Fix rotated nms (#668) 2020-11-25 14:13:26 +08:00
BigBigDream a978764883
Add box_iou_rotated, ml_nms_rotated and nms_rotated (#625)
* add box_iou_rotated, ml_nms_rotated and nms_rotated

* fix lint

* fix lint

* fix .py lint

* fix cpp lint

* add newline at the end

* add new line

* fix unittest

* config google style

* fix lint

* lint

* lint

* yapf

* update

* fix lint

* fix lint

* fix lint

* fix

* fix format

* fix format

* add modified from

* add docstring and update others

* update docstring

* update docstring

* update

* fix bug

* fix bug

* fix bug

Co-authored-by: Cao Yuhang <yhcao6@gmail.com>
2020-11-25 11:00:30 +08:00
RunningLeon f61bb642e5
[Enhance]: Better support for pytorch2onnx for detection models (#648)
* [feature]: Support convertion from Pytorch to ONNX for detection models

* [fix] RuntimeError with integer divide when converting to onnx

* Resolve pylint error

* add comment
2020-11-24 20:04:00 +08:00
wang shiguang 23ecd83d59
Add parrots ci (#662)
* add parrots ci

* update with ops build

* Update build.yml

* cridents env

* Update build.yml
2020-11-24 15:10:24 +08:00
Wenwei Zhang d2fa1abc53
Add documentations for registry (#656)
* add documentations for registry

* update example

* update sphinx

* add intro

* resolve comments
2020-11-23 15:29:22 +08:00
Jerry Jiarui XU 987cb58341
[Enhance] Refactor logger (#659)
* [Enhance] Refactor logger

* fixed test

* make commit optional

* remove debug info

* fixed test
2020-11-23 10:50:18 +08:00
Wenwei Zhang dfa36dfe3b
Merge pull request #652 from dreamerlin/3d
[Feature] Add 3D support in wrapper
2020-11-20 16:33:54 +08:00
yamengxi ec43b671ab
[Enhance]enhance hsigmoid (#657)
* enhance hsigmoid

* delete int
2020-11-19 21:26:26 +08:00
dreamerlin 1a12ac753a fix 2020-11-18 23:03:49 +08:00
dreamerlin e70dca8f86 add conv3d 2020-11-18 21:24:20 +08:00
dreamerlin 144e7567a7 use pytest.mark.parametrize 2020-11-18 15:35:21 +08:00
dreamerlin 86d9f4684a add MaxPool3d 2020-11-15 19:07:46 +08:00
dreamerlin 8ccea20234 add ConvTranspose3d 2020-11-15 18:56:02 +08:00
Jintao Lin 1290bdd181
update docstring (#651) 2020-11-15 18:48:03 +08:00
q.yao c390e327fa
add onnx support to roi_align and roi_pool (#593)
* add onnx support to roi_align and roi_pool

* add test on roi_align and roi_pool

* fix conflict with master
2020-11-14 19:56:35 +08:00
Jiang Yinzuo b3e5a74d22
update README.md for torch 1.7 installation (#642)
* update README.md for torch 1.7 installation

* add cuda 11.0

Co-authored-by: Kai Chen <chenkaidev@gmail.com>
2020-11-14 12:05:57 +08:00
yamengxi c6c230df1b
[Feature] Add CLAHE method (#647)
* add CLAHE

* add CLAHE

* restore

* Add docstring

* modify docstring

* modify CLAHE to clahe

* fix syntax error

* simplify assert

* simplify assert

* add assert test

* fix unittest bug

* fix syntax bug

* fix assert bug
2020-11-11 22:34:14 +08:00
yamengxi d9ef9dabe2
[Feature] Add LUT transform (#646)
* add lut_transform

* restore

* fix test bug

* add test

* Fix syntax bug

* Delete test id

* modfiy docstrinng

* simplify assert
2020-11-11 17:32:41 +08:00
Y. Xiong 1d5678c9fc
bump version to 1.2.0 (#644) 2020-11-09 13:01:03 +08:00
Jerry Jiarui XU bb06f354d5
[Feature] Add hook msg for checkpoint hook (#635)
* [Feature] Add hook msg for ckpt_hook

* [Feature] Add hook msg for ckpt_hook

* add test
2020-11-06 19:56:50 +08:00
Ming-Hsuan-Tu 9a3f1249e5
[fix] Missing arguments when converting dcn to onnx (#624)
* fix issues when converting deformable convolution to onnx

* keep  and  for interface consistency

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2020-11-06 13:01:28 +08:00
Kuro Latency bcf85026c3
Update lr_updater.py (#574)
* Update lr_updater.py

since epoch/iteration in  runner starts with 0, we shouldn't leave the latter iteration to former (12th epoch for example, with first period equal to 12) period.

* Update lr_updater.py

* Update test_hooks.py
2020-11-05 22:20:37 +08:00
Wang Xinjiang 66e249a066
Fix logger (#637)
* Fix logger iter

* Fix logger iter
2020-11-05 13:05:05 +08:00
Wang Xinjiang b74aa7921b
Move mode str to the beginning (#634) 2020-11-03 21:25:54 +08:00
Cao Yuhang 2d52809cdb
Add dynamic scale (#585)
* add dynamic scale

* add type check of loss scale

* fix lint

* minor fix
2020-10-31 19:00:14 +08:00
Kai Chen 03214fd446
Support pytorch 1.7 and update the CI (#631)
* support pytorch 1.7 and update the CI

* fix CI

* fix onnxruntime version in macos
2020-10-31 17:58:15 +08:00
Wang Xinjiang 0b8acd2bae
Use config_file to get config dict in PaviLogger (#630)
* Add 'config' keyword in meta for pavi logger

* Use config file
2020-10-29 17:32:49 +08:00
Y. Xiong f4de390b3c
bump version to 1.1.6 (#629) 2020-10-28 19:38:04 +08:00
Ruohui Wang 761f725b70
Misc updates regarding Windows and unittests (#586)
* SKIP bbox test without CUDA

* use full filepath for creating latest.pth
2020-10-28 16:23:19 +08:00
David de la Iglesia Castro 2e6c8ec803
Refactor logger hooks (#605)
* Refactor tags for consistency

* Fix missing runner

* Fix missing runner

* Fix missing runner

* Fix missing runner

* Fix momentum runner hook inner iter

* Fix tests

* pre-commit run
2020-10-28 12:43:50 +08:00
robin Han 23b2bdbf52
add unittest for onnx convert (#608)
* add unittest for onnx convert

* build onnx and onnxruntime in CI

* skip onnx op unit test while using CUDA

* fix offset==0 case in NMS

* remove tmp file used in test

* delete tmp file before assert so that we can remove the tmp file anyway
2020-10-26 11:33:35 +08:00
Rui Xu 65a60a3d7d
fix iter-time-hook bug when meeting iter-based runner (#623) 2020-10-26 11:26:58 +08:00
David de la Iglesia Castro c8146cc52b
Skip opencv requirement if it's already installed in the env (i.e. via conda) (#616)
* Skip opencv requirement if it's already installed in the env

* pre-commit

* Check opencv conda version

* Check opencv conda version
2020-10-17 17:58:25 +08:00
Jintao Lin a260a96a0f
import_modules_from_strings when loading cfg from file (#606)
* import_modules_from_strings when loading cfg from file

* add unittest to tell whether the feature is enabled as expected

* minor

* set an environment variable instead of writing a file

* use 'shutil' instead of 'os.system'
2020-10-15 22:29:24 +08:00
Ruohui Wang 460959573b
Fix a typo in warning message (#615)
"please use MMDistributedDataParallelinstead." -> "please use MMDistributedDataParallel instead."
2020-10-15 15:08:42 +08:00
Wang Xinjiang 50a33950a4
Add Sync buffer in CheckpointHook (#588)
* Add Sync buffer in CheckpointHook

* add reduce_params in fp16_utils.py

* change default value of sync_buffer to False

* Add world size check

* reset sync_buffer to false

* fix world_size

* Move dist functions into dist_utils.py

* fix small bugs

* Deprecation compatibility

* Change according to comments
2020-10-15 11:02:41 +08:00
Ryan Li 54c527acd5
fix wrappers when using parrots (#613)
* fix wrappers when using parrots

* linting

* refactor according to review
2020-10-13 17:54:16 +08:00
robin Han 993da2bbd7
add switch for onnx exporter (#607) 2020-10-11 18:11:52 +08:00
Rui Xu 0dfc6ae6e6
[bug] fix bug in resuming optim-dict when using epoch based runner (#610)
* fix bug in resuming optim-dict

* add raise error when meeting unexceptable type

* add raise error when meeting unexceptable type
2020-10-11 18:11:33 +08:00
Kai Chen 665fee2441
bump version to 1.1.5 (#603) 2020-10-07 23:09:52 +08:00
Wenwei Zhang 005c408748
Fix wrappers version comparison (#602)
* add version check in wrappers

* fix assersion

* use digital version for version comparison

* fix unit tests

* reformat

* fall back to compare the first two version

* fix unittest

* fix unittest

* fix unit test

* clean unnecessary change
2020-10-06 12:44:56 +08:00
Cao Yuhang fe83261b94
revert dcn (#597) 2020-10-02 00:03:55 +08:00
Kai Chen f3a2be99b9
lint markdown files (#592) 2020-09-29 10:44:44 +08:00
Kai Chen 9141d91ddc
bump version to 1.1.4 (#591) 2020-09-29 00:09:49 +08:00
Cao Yuhang f6d5b0f8cb
fix lint #580 (#590) 2020-09-28 21:52:27 +08:00
wdmwhh c80e4cae54
Fixed fp16_optimizer state bug (#580) 2020-09-28 21:48:56 +08:00
GT9505 acee61d7c5
register deconv in CONV_LAYERS (#582)
* register deconv in CONV_LAYERS

* use ConvTranspose2d implemented in MMCV

* remove repetitive register_module

* update

* add unittest for deconv
2020-09-27 17:51:00 +08:00