在使用lr_image = nn.functional.interpolate(hr_image, scale_factor=1 / 4, mode='bicubic')
报了这个错
因为hr_image是一个三维张量,没有batch_size维度,只用给hr_image通过unsqueeze(0)增加一个batch_size维度就ok了。
在使用lr_image = nn.functional.interpolate(hr_image, scale_factor=1 / 4, mode='bicubic')
报了这个错
因为hr_image是一个三维张量,没有batch_size维度,只用给hr_image通过unsqueeze(0)增加一个batch_size维度就ok了。