Commit Graph

502 Commits (f3a2be99b982b533cb0f916278f7da742c5899ef)
 

Author SHA1 Message Date
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
Wenwei Zhang 34127b9f14
[enhance]: show grad norm in fp16 optimizer hook (#584) 2020-09-27 00:40:39 +08:00
Jerry Jiarui XU 71952ff3aa
[Enhance] Switch to https://download.openmmlab.com (#583) 2020-09-27 00:20:13 +08:00
Wenwei Zhang ed2887bb72
Support to specify LR of DCN's conv_offset (#344)
* Support to specify LR of DCN's conv_offset

* Resolve comments & add unit test

* Resolve formats

* Fix CI for DCN

* Mock DCN when cpu only

* Use mock for cpu testing

* Fix docstring and support ModulatedDCN

* set offset_lr_mult as dcn's arguments, link CU-49u01p

* fix lr bug

* fall back to set LR in constructor

* resolve comments
2020-09-25 20:51:58 +08:00
q.yao 467b4883b9
add torchvision roi_align with aligned=True (#581)
* add torchvision roi_align with aligned=True

* fix for lint test

* fix for lint test2

* format use yapf
2020-09-25 20:19:37 +08:00
David de la Iglesia Castro 6b52e9b55f
Add runner builder (#570)
* Add build_runner

* Parametrize test_runner

* Add imports to runner __init__

* Refactor max_iters and max_epochs from run to init

* Add assertion error messages

* Add test_builder

* Make change retro-compatible

* Raise ValueError if max_epochs and max_iters
2020-09-25 10:25:29 +08:00
Wenwei Zhang 2bb1160e6c
[fix]: fix wrapper comparison of pytorch version (#572) 2020-09-25 00:09:58 +08:00
David de la Iglesia Castro afb73995b9
Add missing by_epoch arg (#576) 2020-09-25 00:00:27 +08:00
Zhiyuan Chen 50af00991c
merge the calling of train/val_step and batch_processor into run_iter (#553)
* merge train/val_step and batch_processor into run_iter

* make self.train_mode of runner as an argument

* remove abstract methods of run_iter() in base_runner
2020-09-22 10:40:00 +08:00
Kai Chen 8c70df3310
bump version to 1.1.3 (#569) 2020-09-21 17:27:21 +08:00
Ryan Li 5cad35bd2d
fix deprecated wrappers exiting bug (#567) 2020-09-18 19:54:37 +08:00
robin Han cc332b26e1
add switch for onnx exporter (#564) 2020-09-18 13:54:00 +08:00
ychan 2114356826
fix dcon forward and backward bug (#565) 2020-09-18 13:48:21 +08:00
ychan b7af0e9ff7
fix mdconv backward bug (#563) 2020-09-17 14:25:16 +08:00
ychan ece32796c3
fix dcn forward bug (#562) 2020-09-17 14:25:05 +08:00
David de la Iglesia Castro 779f47bab9
Allow type to be default arg (#558)
* Add test case for type defined using default_args

* Refactor build_from_cfg

* Update exception of missing type

* pre-commit

* Fix default_args is None

* pre-commit

* Bring back test

* Update exception raising
2020-09-16 19:16:49 +08:00
Kai Chen 7ef3a5e925
Use copy instead of symlink on windows (#557)
* use copy instead of symlink on windows

* use platform.system() instead of sys.platform
2020-09-16 19:12:18 +08:00
Ryan Li 9ecd6b0d5f
Add 'wrap_fp16_model' to __init__.py (#555)
* add 'wrap_fp16_model' to __inti__.py

* add 'wrap_fp16_model' to __init__.py
2020-09-15 20:44:35 +08:00
Qiaofei Li a0cc5a8450
Supports brightness and contrast augmentations (#546)
* add brightness and contrast augmentation

* remove unnecessary

* reformat

* relax the precision constrain for adjust_brightness aug

* fix percision assertion error in unit test

* remove toy

* rename alpha as factor

* use np.testing.assert_allclose in place of np.less_equal
2020-09-14 12:14:43 +08:00
Wenwei Zhang c937d3953d
Mv wrappers into bricks and use wrappers in registry (#550)
* Mv wrappers into bricks and use wrappers in registry

* resolve import issues

* fix import issues

* set nn op forward to torch 1.6.1

* fix CI bug and add warning

* Fix CI by using patch mock

* mv warnings inside deprecated module's initialization
2020-09-12 21:59:17 +08:00
Cao Yuhang c054a2393e
remove assertion (#549) 2020-09-12 16:51:44 +08:00
Xiaojie Li 49e32c2688
Implementation of 2D convolution in tensorflow with `padding` as "same" (#529)
* update impad

* fix docstring

* add shape for impad

* fix unit test

* remove old version & fix doc

* fix linting

* fix doc

* add linear decay learning rate scheduler

* fix impad

* fix setup.cfg

* fix linting

* add yapf

* add swish

* fix lr_updater

* fix lr_updater.py

* update swish

* add swish

* fix inplace

* fix typo

* update

* add same padding

* fix docstring

* add unittest

* fix register

* change name

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-09-09 23:57:48 +08:00
Hongkai Zhang d7c895a361
support multiple interpolation modes for imrotate (#545)
* support multiple interpolation modes for imrotate

* reformat code
2020-09-09 23:54:01 +08:00
su c8435966ed
Reordered the hooks and use attributes rather than args. (#544)
* Reordered the hooks and use attributes rather than args.

Formated.

* Reordering may cause conflict, assign the value first than update such as max_iter

Rewind back the order.

* Rewind back to just use attributes, the update of max_iter and stuff will be done in new hooks.

Minor format.
2020-09-08 18:00:33 +08:00
Qiaofei Li a59a35bcce
Supports Equalize Augmentation. (#543)
* add equalize augmentation in mmcv

* delete unnecessary

* reformat

* remove clip in implementing equalize, and add uint test with case step=0

* remove clip in implementing equalize, and add uint test with case step=0

* add comments for unit test

* rename function name as imequalize
2020-09-08 14:44:14 +08:00
Qiaofei Li 95417a5d24
Supports Color augmentation. (#542)
* add Color augmentation

* reformat

* reformat

* reformat docstring

* reformat docstring

* add more unit test

* add more unit test

* add clip value and uint test for image with type float

* rename function name
2020-09-08 13:09:06 +08:00
Jintao Lin fec7cd61d6
Support reading video from url (#531)
* add url support for VideoReader

* add a comment

* add unittest

* use a connectable url for ci
2020-09-07 00:52:33 +08:00
Qiaofei Li 9769024fbd
Add Translate augmentation. (#538)
* add imtranslate

* add imtranslate

* update comments

* reformat
2020-09-04 20:30:42 +08:00
Wenwei Zhang c6987937fe
[fix] Fix compilation bug in windows and add instruction for windows (#540)
* Add windows instruction and fix compilation bug

* reformat codebase
2020-09-04 15:08:12 +08:00
Wenwei Zhang 16071ed011
[Fix]: fix missing pad_val in impad_to_multiple (#539) 2020-09-04 15:05:14 +08:00
Kai Chen 0664f9b3ed
bump version to 1.1.2 (#537) 2020-09-03 21:30:13 +08:00
Wang Xinjiang fe07f09e0a
Add more default types for json dump (#536) 2020-09-03 19:28:45 +08:00
Qiaofei Li 8dfe58d197
Fix TypeError: Scalar value for argument borderValue is longer than 4 (#535)
* fix TypeError: Scalar value for argument borderValue is longer than 4

* add comments for imshear with case that border_value has more than 3 elements
2020-09-03 17:14:49 +08:00
Jerry Jiarui XU cfd337bb22
Support load with mmcls:// (#511)
* [Feature] Support load models from mmcls

* [Feature] Support load with mmcls://

* hard-code load mmcls

* fixed wrong commit

* add json

* remove cifar
2020-09-02 19:27:47 +08:00
Qiaofei Li 06556c8459
Add shear augmentation (#526)
* Add shear augmentation

* Update geometric.py

* Update geometric.py
2020-09-02 10:57:36 +08:00
Wang Xinjiang 15b37b0b78
Allow imshow_det_bboxes to return image with bboxes. (#527) 2020-09-02 00:39:57 +08:00
Wenwei Zhang fc4993cdec
[feature] Use cu92 & ubuntu1604 for torch 1.3.1 (#524)
* [feature] Use cu92 & ubuntu1604 for torch 1.3.1

* [fix]Use ubuntu-16.04 rather than latest for cu92

* [fix]: soft link cuda path

* Debug cu101 bug

* Update usr/include for cublas with cu101

* Check default nvcc version

* rm :

* add cuda root

* Try to locate cublas_v2.h

* Set cublas version

* Force cublas version

* check usr/local/cuda-10.2

* Add tree

* cp cublas files

* cp cublas files to lib64

* do not tree cuda

* fix path cp error

* cp dir

* Keep using CUDA10.1 only

* recover empty line
2020-09-01 16:15:03 +08:00
Xiaojie Li c3d8eb34ff
add Swish activation (#522)
* update impad

* fix docstring

* add shape for impad

* fix unit test

* remove old version & fix doc

* fix linting

* fix doc

* add linear decay learning rate scheduler

* fix impad

* fix setup.cfg

* fix linting

* add yapf

* add swish

* fix lr_updater

* fix lr_updater.py

* update swish

* add swish

* fix inplace

* fix typo

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-27 00:39:17 +08:00
Kai Chen 66a38c86f8
Move unit tests to specific folders (#520)
* move unit tests to specific folders

* fix path error

* remove some assertions

* fix ignore path
2020-08-25 20:11:56 +08:00
Kai Chen 89e1716afe
Bump version to v1.1.1 (#518)
* bump version to 1.1.1

* add more version utils

* move parse_version_info to version.py
2020-08-24 16:30:45 +08:00
Wang Xinjiang 09b7d6c7dd
Import modules from a string list (#514)
* Custom imports

* Resolve comments

* Add unittest

* Add unittest

* Rename custom_imports to import_modules_from_strings

* Move import_modules_from_strings ito misc.py and allow failed imports

* small change

* small change

* change mmcv.runner to os.path
2020-08-24 13:45:34 +08:00
Kai Chen 270e470ee1
[feature] Add collect_env() to collect environment info (#517)
* [feature] Add collect_env() to collect environment info

* fix unit tests

* fix the case when ops are not compiled

* fix docstring
2020-08-24 13:41:54 +08:00
Matthew Dawkins aebdcb6661
Fix inconsistent return types in pybind11 function prototypes (#509)
* Fix function types

* Remove const
2020-08-24 12:46:04 +08:00
Wenwei Zhang f4a5446e15
Support to split batched_nms when box number is too large (#516)
* Support to split batched_nms when box number is too large

* mv data from gpu to cpu

* Set split_thr through nms_cfg

* clean code

* Update motivation in docstring

* fix typos
2020-08-24 00:32:39 +08:00
Jerry Jiarui XU 83d9a9c89b
[Feature] Add diagonal flip (#515) 2020-08-22 14:29:27 +08:00
Matthew Dawkins eb0414f495
Fix windows compile issues (#510) 2020-08-22 14:24:16 +08:00