블랙 아이스의 학습 데이터셋 구축을 위한 Pix2Pix 기반 데이터 증강 방안 (Pix2Pix-Based Data Augmentation Method for Building an Image Dataset of Black Ice)

서지정보

· 발행기관 : 대한교통학회
· 수록지 정보 : 대한교통학회지 / 40권 / 4호 / 539 ~ 554페이지
· 저자명 : 이호준, 강민희, 송재인, 황기연

초록

최근 인공지능 분야는 급속도로 성장하고 있으며, 딥러닝 기법을 활용한 이미지 인식 기술이 인간의 인식률을 뛰어넘는 수준으로 진화하였다. 이는 대용량의 데이터셋을 활용한 학습이 선행 되어야 함에 따라 CIFAR-10, Fashion-MNIST, MS-COCO, Cityscapes와 같은 대용량 오픈 데이터셋을 공개하는 등 데이터셋을 구축하기 위한 노력이 이루어지고 있다. 그러나, 데이터 수집 과정에서 많은 시간과 비용이 소모됨에 따라 데이터 증강 기술을 활용하고 있으며, Autoencoder, 생성적 적대 신경망 등 딥러닝을 활용한 데이터 증강 연구가 활발하게 이루어지고 있다. 특히 교통 분야에서는 교통사고를 예방하기 위해 이미지 인식 기술과 데이터 증강 기법을 활용하고 있지만, 치사율이 높고 대형 사고로 이어지는 블랙 아이스의 경우 데이터 구득에 한계가 존재하므로 대용량 데이터셋을 구축하는 것에 어려움을 겪고 있다. 이에 본 연구에서는 Pix2Pix를 활용한 블랙 아이스 이미지 데이터의 증강 방안을 제시한다. Pix2Pix 모델의 학습을 위해 데이터 수집과 전처리를 진행하였으며, 모델 학습을 통해 새로운 도로에 생성된 블랙 아이스 이미지를 확보하였다. 학습 결과 분석에서는 loss와 Frechet Inception Distance score를 도출하는 정량적 평가와 생성된 이미지를 활용하여 정성적 평가도 수행하였으며, 이를 통해 블랙 아이스 이미지를 성공적으로 생성하는 것을 확인하였다. 본 연구는 Pix2Pix를 활용하여 도로 위에 생성된 블랙 아이스 이미지의 생성이 가능한 것을 확인하였으며, 이를 통해 블랙 아이스 데이터셋을 구축하여 향후 이를 활용한 객체 검출 연구가 수행되어야 할 것으로 판단된다.

영어초록

Recently, the field of artificial intelligence is growing rapidly, and image recognition technology using deep learning technology has evolved to a level that exceeds human recognition rate. To this end, it is essential to build large-capacity datasets, and efforts are being made to secure datasets by disclosing large-capacity open datasets such as CIFAR-10, Fashion-MNIST, MS-COCO, and Cityscapes. However, as data collection takes a lot of time and money, data augmentation technology is being used, and data augmentation research using deep learning such as Autoencoder and Generative Adversarial Network is being actively conducted. In particular, image recognition technology and data enhancement techniques are used to prevent traffic accidents, but black ice, which has a high mortality rate and leads to large accidents, has limitations in data acquisition, making it difficult to build large datasets. Therefore, this study proposes a method for enhancing black ice image data using Pix2Pix. Data collection and preprocessing were conducted to learn the Pix2Pix model, and black ice images created on new roads were secured through model learning. In the analysis of learning results, a quantitative evaluation was performed to derive loss and Frechet Inception Distance score, and a qualitative evaluation was also performed on the generated image. As a result of the learning, loss and Frechet Inception Distance score decreased as the learning progressed, and it was confirmed that the black ice image generated on the road was successfully generated through qualitative evaluation. This study confirmed that it is possible to generate a black ice image generated on the road using Pix2Pix, and through this, it is judged that object recognition research using the dataset should be conducted in the future by constructing a black ice dataset.

참고자료

· 없음