이제 데이터 전처리에서 사용할 변수에 대한 값을 정의합니다.
코드 6-67 변수에 대한 값 정의
size = 224
mean = (0.485, 0.456, 0.406)
std = (0.229, 0.224, 0.225)
batch_size = 32
torchvision.datasets.ImageFolder를 이용하여 훈련과 테스트 데이터셋을 불러옵니다. 데이터셋은 개와 고양이에 대한 예제를 계속 사용합니다. 추가 학습을 위해 캐글에서 이미지 데이터셋을 내려받아 학습해도 좋습니다. 혹은 이미지를 찾기 어렵다면 책에서 사용하는 이미지 데이터셋을 이용해도 좋습니다.
코드 6-68 훈련과 테스트 데이터셋 불러오기
cat_directory = r'../chap06/data/dogs-vs-cats/Cat/'
dog_directory = r'../chap06/data/dogs-vs-cats/Dog/'
cat_images_filepaths = sorted([os.path.join(cat_directory, f) for f in os.listdir(cat_directory)])
dog_images_filepaths = sorted([os.path.join(dog_directory, f) for f in os.listdir(dog_directory)])
images_filepaths = [*cat_images_filepaths, *dog_images_filepaths]
correct_images_filepaths = [i for i in images_filepaths if cv2.imread(i) is not None]