반응형 컴퓨터비전(CV)/논문 구현2 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. [논문 구현]Fully Convolutional Networks for Semantic Segmentation(FCN) https://github.com/327aem/paper_review/tree/main/Segmentation GitHub - 327aem/paper_review: 논문 리뷰/구현 논문 리뷰/구현. Contribute to 327aem/paper_review development by creating an account on GitHub. github.com import torch import torch.nn as nn import torchvision.models as models class FCN16(nn.Module): def __init__(self,model,hidden =64,kernel_size=3,padding = 1, num_classes = 21): super(FCN16,self).__.. 2022. 3. 16. 이전 1 다음 반응형
