Wang0812
Wang0812
print(image.size()) # (C，H, W), 通道顺序（R,G,B) unloader = transforms.ToPILImage() def tensor_to_PIL(tensor): image = tensor.cpu().clone() image = image.squeeze(0) image = unloader(image) return image model = CSNet(32, 0.1) res = model(image) net =...
很高兴收到您的回复，我刚刚尝试了save，但是保存的图片是一片空白，请问有什么解决办法吗，不知道是哪里出了问题，学习
import matplotlib.pyplot as plt from matplotlib.image import imread from net import CSNet import torchvision import numpy as np from Data import calculate_valid_crop_size from PIL import Image from torchvision import transforms...
我用的是测试的那几张图像，512x512的，想不明白，请问可以传一份保存图像的代码吗，非常感谢