본문으로 바로가기

이미지 확대 프로그램 'waifu2x-caffe'

category IT/팁 2016.05.26 21:04

이 프로그램은 이미지 변환 소프트웨어인 'waifu2x'의 변환 기능만을 Caffe를 이용하여 재작성, 윈도우 용으로 컴파일한 프로그램입니다. 한글화가 되어있고 GUI이기 때문에 사용하기 쉽다는 장점이 있습니다. 물론 CUI 모드로도 사용할 수 있습니다. 또한 CPU가 아닌 GPU, 그중에서 NVIDIA 그래픽카드가 사용하는 CUDA 가속을 이용해 이 작업을 빠르게 할 수도 있습니다.


다운로드는 GitHub(링크)에서 받으실 수 있습니다. 해당 링크의 릴리즈 버전의 맨 끝의 'Downloads' 탭에서 'waifu2x-caffe.zip'파일을 다운로드하면 됩니다.



사용 방법

1. '입력 경로'의 '찾기'를 눌러 이미지를 선택한다.

2. '출력 경로'의 '찾기'를 눌러 변환된 이미지가 저장되기 원하는 폴더를 선택한다.

3. '확대 크기'를 변경해 원하는 배율 혹은 크기를 선택한다.

4. '모델'에서 자기가 변환하고자 하는 이미지에 맞는 모드를 선택한다.

5. 'App Setting'에서 자기가 사용하려고 하는 프로세서를 선택한다

5-1. CPU를 사용할 경우 'Use Processor' 박스 안의 CPU를 선택한다.

5-2. NVIDIA 그래픽카드를 사용할 경우 'CUDA (cuDNN 사용)'을 선택한다. (※ cudnn64_5.dll이 없다는 에러가 뜰 경우 하단 참조)

6. '시작'을 누른다.



결과

<원본 파일 - 600x285, 119.5KB>


<변환 파일(Photo 모델) - 1200x570, 844.2KB>



Q&A

Q. TTA 모드가 뭔가요?

A. Test-Time Augmentation의 약자로 사용할 경우 변환이 8배 느려지지만 PSNR(이미지 평가 지수)가 0.15정도 오른다고 합니다.


Q. 분할 크기는 뭔가요?

A. 프로그램이 한번에 처리하는 크기입니다. 이 분할 크기가 이미지 가로/세로 크기의 약수일 경우 속도가 빨라집니다. 너무 크게 잡을 경우 컴퓨터의 메모리 용량을 초과할 수도 있으니 조심하세요.


Q. 모델이 뭔가요?

A.    2D 일러스트 (RGB 모델) : 이미지의 모든 RGB를 변환시키는 2D 일러스트용 모델.

사진 · 애니 (Photo 모델) : 사진 · 애니메이션용 모델.

2D 일러스트 (UpRGB 모델) : 2D 일러스트 (RGB 모델)보다 빠르고, 그 이상의 화질로 변환시키는 모델. 다만 RGB 모델보다 소비하는 메모리(VRAM)의 양이 많기 때문에 변환 중에 강제로 종료될 경우 분할 크기를 조절할 것.

사진 · 애니 (UpPhoto 모델) : 사진 · 애니 (Photo 모델)보다 빠르고, 그 이상의 화질로 변활시키는 모델. 다만 RGB 모델보다 소비하는 메모리(VRAM)의 양이 많기 때문에 변환 중에 강제로 종료될 경우 분할 크기를 조절할 것.

2D 일러스트 (Y 모델) : 이미지의 밝기(Y)만을 변환시키는 2D 일러스트용 모델.



Q. cudnn64_5.dll이 없다는데요?

A. 엔비디아의 cuDNN(링크, 회원가입 필요)에서 v3 이상을 다운받아 waifu2x-caffe 폴더에 넣으면 됩니다. cudnn64_5는 다운로드 받은 cuDNN의 bin 폴더에 있습니다.


댓글을 달아 주세요

  1. BlogIcon sdeki 2016.11.10 02:12 신고

    설명 잘 읽고 갑니다. 모델별 무슨 차이가 있나 궁금했었는데 덕분에 풀렸네요.

  2. 잔손 2018.04.09 09:26 신고

    좋은 설명 감사합니다. 덕분에 프로그램에 쉽게 다가갈 수 있게 되었어요.