这个函数刚开始用,还不太理解,后来看了下说明,大概明白了
input是一个四维向量,包括batch*depth*h*w
然后,size就是一个两个数组成的Tuple,对应的是input尺寸的h和w
res=torch.nn.functional.interpolate(input,(orig_img_h,orig_img_w),mode='bilinear')
这个函数刚开始用,还不太理解,后来看了下说明,大概明白了
input是一个四维向量,包括batch*depth*h*w
然后,size就是一个两个数组成的Tuple,对应的是input尺寸的h和w
res=torch.nn.functional.interpolate(input,(orig_img_h,orig_img_w),mode='bilinear')