① 이미지 데이터에 대한 레이블 값(dog, cat)을 가져옵니다.

ⓐ img_path: 이미지가 위치한 전체 경로를 보여 줍니다(필자가 사용한 경로는 다음과 같지만 환경에 따라 경로는 다를 수 있습니다).

예 E:/torch/chap06/data/dogs-vs-cats/Dog/dog.113.jpg

이때 이미지 데이터의 레이블(dog)을 가져오려면 ‘/’와 ‘.’을 제거해야 합니다. 다음의 ⓑ와 ⓒ가 ‘/’, ‘.’을 제거하기 위해 사용됩니다.

ⓑ img_path.split('/')[-1]: 이미지의 전체 경로에서 ‘/’를 제거합니다. 제거 결과는 다음과 같습니다.

예 dog.113.jpg

ⓒ split('.')[0]: 마지막으로 img_path.split('/')[-1]을 통해 얻은 결과인 ‘dog.113.jpg’에서 ‘.’을 제거합니다. 이때 ‘dog.113.jpg’를 ‘.’을 기준으로 분리시켰고, 분리된 값들에서 첫 번째 값([0])을 가져오면 결과는 다음과 같습니다.

예 dog

전처리에서 사용할 변수에 대한 값을 정의합니다.

코드 6-7 변수 값 정의

size = 224
mean = (0.485, 0.456, 0.406)
std = (0.229, 0.224, 0.225)
batch_size = 32