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
