卷积神经网络常用函数Pytorch

上下采样函数--interpolate

参考链接：https://zhuanlan.zhihu.com/p/166323682

x = nn.functional.interpolate(x, scale_factor=8, mode='bilinear', align_corners=False)

solov2
kernel_feat = F.interpolate(kernel_feat, size=seg_num_grid, mode='bilinear') #上采样 成 [1, seg_num_grid, seg_num_grid]

Pytorch中view()、squeeze()、unsqueeze()、torch.max()函数经常用：

1、view()函数作用是将一个多行的Tensor,拼接成一行
2、squeeze(0)表示如果第一维度值为1，则去掉，否则不变。
3、unsqueeze()与squeeze()作用相反，即增加一维
4、torch.max()返回两个结果，第一个是最大值，第二个是对应的索引值；第二个参数 0 代表按列取最大值并返回对应的行索引值，1 代表按行取最大值并返回对应的列索引值。

torch.contiguous()方法

参考链接：https://blog.csdn.net/qq_37828380/article/details/107855070