그럼 이제 파이토치로 ResNet을 구현해 보겠습니다. 이번 예제는 앞에서 살펴보았던 LeNet, AlexNet 등과는 다르게 상당히 복잡하기 때문에 네트워크 구성과 관련하여 주의 깊게 살펴보기 바랍니다.
언제나 그렇듯 필요한 라이브러리를 호출합니다.
코드 6-65 필요한 라이브러리 호출
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import torch.utils.data as data
import torchvision
import torchvision.transforms as transforms
import torchvision.datasets as datasets
import torchvision.models as models
import matplotlib.pyplot as plt
import numpy as np
import copy
from collections import namedtuple ------ ①
import os
import random
import time
import cv2
from torch.utils.data import DataLoader, Dataset
from PIL import Image
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')