search

PANPEIWEN / ABC

ABC: Attention with Bilinear Correlation for Infrared Small Target Detection
MIT License
29 stars 3 forks source link

关于代码问题 #4

Open orange998 opened 1 year ago

orange998 commented 1 year ago

self.Convtans2 = ConvTransformer(filters[0], filters[1], pow(features[0], 2), features[0]) 想问一下在做Convtans2函数的时候为什么使用pow(features[0], 2)，这样做的意义是什么

PANPEIWEN commented 1 year ago

self.Convtans2 = ConvTransformer(filters[0], filters[1], pow(features[0], 2), features[0]) 想问一下在做Convtans2函数的时候为什么使用pow(features[0], 2)，这样做的意义是什么

这里我们是为了将HWC变为(HW)C，然后输入到全连接层中，由于在数据预处理的时候将图片H和W resize成一样大小，所以我在这里直接用pow表示