반응형 srgan 분석1 srgan 코드 분석 # https://github.com/eriklindernoren/PyTorch-GAN/blob/master/implementations/srgan/models.py # models.py import torch.nn as nn import torch.nn.functional as F import torch from torchvision.models import vgg19 import math class FeatureExtractor(nn.Module): def __init__(self): super(FeatureExtractor, self).__init__() vgg19_model = vgg19(pretrained=True) self.feature_extractor = nn.Sequential(*list.. 2022. 7. 24. 이전 1 다음 반응형
