FEM模块的三大输入:
1.support_feature, 2.query_feature, 3.Training-Free Piror Mask
一、support_feature:
将resnet50 block2,block3输出的特征拼接起来,然后使用1x1Conv进行降维,将此特征与support对应的mask相乘,然后采用平均池化,特征维度为[bs,c,1,1],c为通道数
保存在supp_feat_list中,列表长度为shot,对应shot个特征
二、Training-Free Piror Mask
将resnet50 block3输出的特征与support mask相乘,然后输入到resnet50 block4中。 输出特征维度为[4, 2048, 60, 60]
保存在final_supp_list中,列表长度为shot,对应shot个特征