# 看tensor batch中的第一张图片
kan = tensor[0:1, :, :, :].squeeze(0).squeeze(0).cpu()
kan = T.ToPILImage()(kan)
kan.show()
从Tensor看图像
最新推荐文章于 2023-09-13 11:07:12 发布
# 看tensor batch中的第一张图片
kan = tensor[0:1, :, :, :].squeeze(0).squeeze(0).cpu()
kan = T.ToPILImage()(kan)
kan.show()