Mr_Lowbee的博客

将pip版本升级到>=10.0.0一行代码pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

方法一import osimport torchos.environ["CUDA_VISIBLE_DEVICES"]="1, 2" 方法二CUDA_VISIBLE_DEVICES=1 python **.py

# 看tensor batch中的第一张图片kan = tensor[0:1, :, :, :].squeeze(0).squeeze(0).cpu()kan = T.ToPILImage()(kan)kan.show()

from PIL import Imageimport torchfrom torchvision import transforms as Tdef add_noise(input_img, noise_sigma): noise_sigma = noise_sigma / 255 noise_img = torch.clamp(input_img+noise_sigma*torch.randn_like(input_img), 0.0, 1.0) return no.

class TVLoss(nn.Module): def __init__(self, TVLoss_weight=1): super(TVLoss, self).__init__() self.TVLoss_weight = TVLoss_weight def forward(self, x): batch_size = x.size()[0] h_x = x.size()[2] w_x = x.size()

from PIL import Imagedata_root = './DIV2K_train_HR'data_save = './DIV2K_gray'for i in range(800): img_index = i + 1 print(img_index) img_rgb = Image.open(data_root + "/" + str(img_index).zfill(4)+'.png') img_gray = img_rgb.convert('L')

from PIL import Imageimport torchimport randomcrop_size = 128# 图片数量为200for i in range(200): print(i) img_index = random.randint(1, 800) # 大图的数量 img = Image.open('./DIV2K_gray' + "/" + str(img_index) + '.png') img_H = img.size[1]