Proceedings of the Korean Society of Computer Information Conference (한국컴퓨터정보학회:학술대회논문집)
- 2022.01a
- /
- Pages.389-390
- /
- 2022
Enhanced Deep Learning for Animal Image Patch Classification
동물 이미지 패치 분류를 위한 향상된 딥 러닝
- Shin, Seong-Yoon (School of Computer Inf. & Communication Eng., Kunsan National University) ;
- Lee, Hyun-Chang (Dept. of Digital Contents Eng, Wonkwang University) ;
- Shin, Kwang-Seong (Dept. of Digital Contents Eng, Wonkwang University)
- Published : 2022.01.12
Abstract
본 논문에서는 동물 이미지 분류를 위한 작은 데이터 세트를 기반으로 하는 향상된 딥 러닝 방법을 제안한다. 먼저, CNN을 사용하여 작은 데이터 세트에 대한 훈련 모델을 구축한다. 데이터 증대를 사용하여 훈련 세트의 데이터 샘플을 확장한다. 다음으로, VGG16과 같은 대규모 데이터 세트에서 사전 훈련된 네트워크를 사용하여 작은 데이터 세트의 병목 현상 기능을 추출한다. 그리하여 두 개의 NumPy 파일에 새로운 훈련 데이터 세트 및 테스트 데이터 세트로 저장한다. 마지막으로 완전히 연결된 네트워크를 훈련시킨다.