def forward(self, z_f, x_f): cls_kernel = self.template_cls_conv(z_f) loc_kernel = self.template_loc_conv(z_f) cls_feature = self.search_cls_conv(x_f) loc_feature = self.search_loc_conv(x_f) cls = xcorr_fast(cls_feature, cls_kernel) loc = self.loc_adjust(xcorr_fast(loc_feature, loc_kernel)) return cls, loc
def forward(self, z_f, x_f): cls_kernel = self.template_cls_conv(z_f) #模板通道数目成倍提升 loc_kernel = self.template_loc_conv(z_f) cls_feature = self.search_cls_conv(x_f) #搜索区域只经过卷积通道数目保持不变 loc_feature = self.search_loc_conv(x_f) cls = xcorr_fast(cls_feature, cls_kernel) #使用分组卷积 loc = self.loc_adjust(xcorr_fast(loc_feature, loc_kernel)) #对于位置分支多加了一个调整分支 return cls, loc
def forward(self, kernel, search): kernel = self.conv_kernel(kernel) search = self.conv_search(search) middle_feat = xcorr_fast(search, kernel) out = self.head(middle_feat) return out