GPU 를 이용한 콘볼루션 뉴럴 네트워크 기반 초해상화 설계 및 구현

Accelerating Deep learning based Super resolution algorithm using GPU

  • 기세환 (한국과학기술원 전기 및 전자 공학과) ;
  • 최재석 (한국과학기술원 전기 및 전자 공학과) ;
  • 김수예 (한국과학기술원 전기 및 전자 공학과) ;
  • 김문철 (한국과학기술원 전기 및 전자 공학과)
  • Ki, Sehwan (Korea Advanced Institute of Science and Technology Dep. Of Electronic Engineering) ;
  • Choi, Jaeseok (Korea Advanced Institute of Science and Technology Dep. Of Electronic Engineering) ;
  • Kim, Sooye (Korea Advanced Institute of Science and Technology Dep. Of Electronic Engineering) ;
  • Kim, Munchurl (Korea Advanced Institute of Science and Technology Dep. Of Electronic Engineering)
  • 발행 : 2017.06.21

초록

본 논문에서는 딥 콘볼루션 신경망 구조를 사용하여 학습된 초해상화 알고리즘을 GPU 프로그래밍을 통해 실시간 동작이 가능하도록 하는 방법을 제시하였다. 딥 러닝이 많이 대중화 되면서 많은 영상처리 알고리즘이 딥러닝을 기반으로 연구가 되었다. 하지만 계산 량이 많이 필요로 하는 딥 러닝 기반 알고리즘은 UHD 이상의 고해상도 영상처리에는 실시간 처리가 어려웠다. 이런 문제를 해결하기 위해서 고속 병렬 처리가 가능한 GPU 를 사용해서 2K 입력영상을 4K 출력 영상으로 확대하는 딥 초해상화 알고리즘을 30 fps 이상의 처리 속도로 동작이 가능하도록 구현을 하였다.

키워드