마지막으로 함수를 호출하여 테스트 데이터셋에 대한 예측 결과를 이미지와 함께 보여 줍니다.

코드 6-38 예측 결과에 대해 이미지와 함께 출력

display_image_grid(test_images_filepaths)

그림 6-12는 모델이 분류한 예측 결과를 보여 줍니다.

▲ 그림 6-12 테스트 데이터셋에 대한 AlexNet 모델의 예측 결과

역시 예측 결과가 좋지 않습니다. 앞에서 언급했지만 데이터셋이 많으면 성능이 좋아질 수 있습니다. 책의 예제들은 실습할 PC(노트북)의 성능을 고려하여 데이터셋을 제한시켰습니다. 성능은 파라미터 튜닝을 통해서도 향상시킬 수 있습니다. 자세한 내용은 ‘8장 성능 최적화’에서 다룹니다.