TEAM EDA

이전글 FCN에 이어서 코드를 한번 살펴보도록 하겠습니다. 기본적인 FCN32s의 코드는 다음과 같습니다. import torch import torch.nn as nn class FCN32s(nn.Module): def __init__(self, num_classes=21): super(FCN32s, self).__init__() self.relu = nn.ReLU(inplace=True) # conv1 self.conv1_1 = nn.Conv2d(3, 64, 3, padding=100) self.relu1_1 = nn.ReLU(inplace=True) self.conv1_2 = nn.Conv2d(64, 64, 3, padding=1) self.relu1_2 = nn.ReLU(inplace=True) ..