今天看CAPS源码时发现在计算x1
在I2
中对应位置的概率分布(对应于论文的3.2 Differentiable Matching Layer)有两种方式correlation
和distance
:
-
correlation
: 将I1
和I2
的 feature mapf1
和f2
作矩阵乘法,然后求softmax作为概率分布p
:
p = s o f t m a x ( f 1 ⋅ f 2 T ) p = softmax(f_1·f_2^T) p=softmax(f1⋅