【算法思考】Pytorch Align Corners 用法总结

Kross Sun

已于 2023-01-29 00:10:17 修改

阅读量1.9k

点赞数 4

分类专栏： AI算法常用技术文章标签： pytorch 算法深度学习

于 2023-01-29 00:06:31 首次发布

本文链接：https://blog.youkuaiyun.com/S_Kross/article/details/128775945

版权

前言

Pytorch的许多接口都需要提供align_corners=True/False。为了更好的使用pytorch提供的接口，有必要了解这个参数所表示的具体含义。下面我们会通过图形化的方式展示。

需要align_corners参数的接口

Pytorch 中使用 align_corners的接口如下所示：

class torch.ao.nn.quantized.functional.interpolate(input, size=none, scale_factor=none, mode='nearest', align_corners=none)
torch.nn.functional.affine_grid(theta, size, align_corners=none)
torch.nn.functional.grid_sample(input, grid, mode='bilinear', padding_mode='zeros', align_corners=none)
torch.nn.functional.interpolate(input, size