mirrors/mmcv
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmcv.git synced 2025-06-03 21:54:52 +08:00
Code Issues Projects Releases Wiki Activity
885 Commits 13 Branches 108 Tags
Commit Graph

2 Commits

Author SHA1 Message Date
Jintao Lin
6ece0e5d19
Add pairwise function for 'gaussian' and 'concatenation' mode in NonLocal. (#424) 
* add pairwise function for 'gaussian' and 'concatenation' mode

* rename test function

* decrease the complexity of nonlocal unittest

* fix typo and make unittest more complete

* add unittest when zero_init is False

* minor fix

* pack theta and phi

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
 2020-07-17 23:44:50 +08:00
Jintao Lin
dcc20f3a4b
Add _NonLocalNd module (#331) 
* add non_local module

* rewrite non local module comments

* perfect docstring and adjust init function

* not to init norm layer

* Correct initialize when there is a norm

* set normal method for both embedded_gaussian and dot_product
 2020-06-18 20:56:48 +08:00