Commit Graph

802 Commits (04daea425bcb0a104d8b4acbbc16bd31304cf168)
 

Author SHA1 Message Date
Wenwei Zhang d5241b7087
Allow the initial value to be set in dynamic mode (#692)
* allow the initial value to be set in dynamic mode

* add logging for grad norm back

* resolve comments
2020-12-09 22:34:37 +08:00
Wenwei Zhang a0902e6de8
allow non-master build of parrots (#702) 2020-12-09 21:28:06 +08:00
Rui Xu cd96a84add
[Feature] Add clamp activation layer. (#685)
* add clamp without unittest

* add clamp-act with unit test

* fix name bug

* use logical and

* fix logical_and

* fix linting

* rename ClampLayer to Clamp

* rename ClampLayer to Clamp

Co-authored-by: nbei <631557085@qq.com>
2020-12-09 10:28:32 +08:00
Wenwei Zhang 1e925a05a2
Skip pat build in PR (#699)
* skip pat build in PR

* rename
2020-12-08 19:19:35 +08:00
Wang Xinjiang c9f96855b0
Allow list index keys in Config.merge_from_dict (#696)
* Allow list keys in Config merge from dict

* Reformat

* Set allow_list_keys default as True in merge_from_dict

* Fix docstring

* fix a small typo
2020-12-08 17:17:27 +08:00
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