더북(TheBook)

코드 6-3 이미지 데이터셋을 불러온 후 훈련, 검증, 테스트로 분리

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]  ------ 개와 고양이 이미지들을 합쳐서 images_filepaths에 저장
correct_images_filepaths = [i for i in images_filepaths if cv2.imread(i) is not None] ------ ②

random.seed(42) ------ ③
random.shuffle(correct_images_filepaths)
train_images_filepaths = correct_images_filepaths[:400] ------ 훈련용 400개의 이미지
val_images_filepaths = correct_images_filepaths[400:-10] ------ 검증용 92개의 이미지
test_images_filepaths = correct_images_filepaths[-10:] ------ 테스트용 열 개의 이미지
print(len(train_images_filepaths), len(val_images_filepaths), len(test_images_filepaths))
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.