pytorch
GY-赵
这个作者很懒,什么都没留下…
展开
-
计算IoU(D2L)
IoU代码实现细节原创 2022-07-30 11:49:04 · 220 阅读 · 0 评论 -
slice
class slice(start, end, step)class slice(stop)arg:demo:输出原创 2022-06-30 10:44:51 · 80 阅读 · 0 评论 -
torch.roll
torch.roll原创 2022-06-28 21:12:38 · 166 阅读 · 0 评论 -
torch stack() \ totch.meshgrid()
tensor([[0, 0, 0, 0],[1, 1, 1, 1],[2, 2, 2, 2],[3, 3, 3, 3]])tensor([[0, 1, 2, 3],[0, 1, 2, 3],[0, 1, 2, 3],[0, 1, 2, 3]])相当于(0,0)(0,1)…两个二维张量相同位置元素对就是一个坐标tensor([[[0, 0, 0, 0],[1, 1, 1, 1],[2, 2, 2, 2],[3, 3, 3, 3]],...............原创 2022-06-25 10:54:41 · 237 阅读 · 0 评论 -
pytorch中的插值算法函数模块--interpolate
#### 官方函数说明torch.nn.functional.interpolate(input, size=None, scale_factor=None, mode='nearest', align_corners=None, recompute_scale_factor=None)根据给定的size或者scale_factor(放缩因子)下采样/上采样 输入mode指定使用的插值算法支持输入为时序、空间或者三维立体等输入,输入应该是3-D、4-D、5-D输入维度格式应该是: mini-b原创 2022-04-24 10:38:20 · 5510 阅读 · 0 评论