GCP概念
协方差矩阵描述的是一组随机变量两两之间的相关性大小。全局协方差池化的做法,就是通过计算特征图的协方差矩阵(二阶信息)来选出这个能代表特征图数据分布情况的值。
torch.cov()函数可以自行实现协方差矩阵的计算,具体计算过程由下图官方解释。
根据协方差的计算公式,简单写代码实现GCP,并应用于CNN网络的分类层中。
class CovariancePooling(nn.Module):
def __init__(self, input_c: int, squeeze_factor: int = 4): # 输入通道数,超参压缩因子(默认为4)
super(CovariancePoolin