bilinear_kernel的python实现

网友投稿 625 2022-10-06

bilinear_kernel的python实现

bilinear_kernel的python实现

# 定义 bilinear kerneldef bilinear_kernel(in_channels, out_channels, kernel_size): ''' return a bilinear filter tensor ''' factor = (kernel_size + 1) // 2 if kernel_size % 2 == 1: center = factor - 1 else: center = factor - 0.5 og = np.ogrid[:kernel_size, :kernel_size] filt = (1 - abs(og[0] - center) / factor) * (1 - abs(og[1] - center) / factor) weight = np.zeros((in_channels, out_channels, kernel_size, kernel_size), dtype='float32') weight[range(in_channels), range(out_channels), :, :] = filt return torch.from_numpy(weight)

版权声明:本文内容由网络用户投稿,版权归原作者所有,本站不拥有其著作权,亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容,请联系我们jiasou666@gmail.com 处理,核实后本网站将在24小时内删除侵权内容。

上一篇:微信小程序通过api接口将json数据展现到小程序(小程序json数据格式)
下一篇:微信小程序城市定位的实现(微信小程序城市定位的实现步骤)
相关文章

 发表评论

暂时没有评论,来抢沙发吧~