UOMOP

원본 +  fftimport osimport torchimport numpy as npfrom torch.utils.data import DataLoader, Datasetimport torch.nn as nnimport torch.optim as optimimport picklefrom sklearn.preprocessing import StandardScalerfrom torchvision import transformsfrom torch.utils.data import DataLoaderfrom InPainting import Loader_maker_for_InPaintingfrom tqdm import tqdmfrom utils import *# 모델 파일 로드dim = 1024snr = 40mo..

import matplotlib.pyplot as pltimport numpy as npimport torchvision.transforms as transformsimport torchvision.datasets as datasetsfrom torch.utils.data import DataLoaderfrom scipy.ndimage import sobel# CIFAR-10 데이터셋 로드transform = transforms.Compose([transforms.ToTensor()])cifar10 = datasets.CIFAR10(root='./data', train=True, download=True, transform=transform)dataloader = DataLoader(cifar10, ba..