[ESRGAN] 삽질 후기

이번 학기에 학교에서 프로젝트 중 ESGAN 모델을 써야하는 프로젝트가 있어서 ESGAN을 사용하다가

삽질을 8328483번 정도 했다.

나중을 위해서 후기를 써보겠다.

 

1. git clone github.com/open-mmlab/mmsr.git

open-mmlab/mmsr

 

2. 사전작업

training 시키기 전에 install 해줘야 할 것들을 다 해주자

 

3. dataset 전처리 과정

https://github.com/open-mmlab/mmsr/blob/master/datasets/DATASETS.md

open-mmlab/mmsr

위 링크에서 시키는 대로 dataset를 가공해야한다..

안그러면 tensor shape error 같은게 뜬다..

물론 mmsr이 제공하는 dataset으로 하면 문제 없지만 나처럼 다른 dataset을 가지고 학습을 진행한다면 가공을 해야한다!!!

 

- 쉽게 정리

1) GT  image 준비 - 해상도 좋은 이미지

2) 그 와 이름이 같은 LR image 준비 - 해상도 낮은 이미지 -> data_scripts/generate_mod_LR_bic.py 파일을 이용하여 만드는 것을 추천..

3) 이미지를 crop하기

4) LMDB file 만들기 

5) 테스트 해보기

 

위 과정들을 완료하여야 training을 진행할 수 있다.

 

4. pretrain model 준비

https://drive.google.com/drive/folders/17VYV_SoZZesU6mbxz2dMAIccSSlqLecY 

ESRGAN_models - Google 드라이브

위 링크에서 pre-trained 모델을 다운받을 수 있다.

 

5. config 파일 수정

1) dataset 경로 수정하기

2) 진짜진짜 중요한거 -> gpu가 우리는 한개였는데 

gpu_ids: [3] 이거를 [0]으로 바꿔줘야한다.... 이거 때문에 2일을 꼬박 날렸다!!!!

 

6. training 해주기

github에서 하라는대로 training을 해주면 된다.

 

이와 같은 과정과 함께라면 별다른 에러없이 진행될 수 있다. 그럼 안농~