torch.nn.functional.interpolate

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/lianbus/article/details/88813567

这个函数刚开始用,还不太理解,后来看了下说明,大概明白了

input是一个四维向量,包括batch*depth*h*w

然后,size就是一个两个数组成的Tuple,对应的是input尺寸的h和w

res=torch.nn.functional.interpolate(input,(orig_img_h,orig_img_w),mode='bilinear')

 

展开阅读全文

没有更多推荐了,返回首页