Fix errors and Add features

Dongwoo-Im commented 2 years ago

What is this PR?

Error 해결, 기능 추가, 표현 수정 등의 코드 공유를 위한 PR입니다.

Changes

Error 해결

our_datasets.py : haze_name을 gt_name으로 바꾸는 과정에서 생기는 에러 해결
finetune.py : model save 시 epoch 기준 잘못된 것 수정

기능 추가

CLAHE.py : clahe parameter를 폴더 이름에 반영
finetune.py : gt 폴더 이름을 의미하는 opt.unlabel_gt_type 추가
finetune.py : debug 모드를 위한 opt.not_train & opt.not_valid 추가
finetune.py : valid 과정에서 input & output image logging 추가

이외에 자잘하게 표현 같은거 수정했습니다.

To reviewers

Dongwoo-Im commented 2 years ago

train label data : MRFID BeDDE train unlabel data : RESIDE_RTTS valid data : RESIDE_SOTS_OUT

파일 이것저것 바뀐 게 많아서 제 branch에 Dehazeformer 모델 파일 가져와서 하시는게 아마 편하실 것 같습니다. 어처피 언젠가 합칠 것 같아서 아예 push 해놓으셔도 상관없을 것 같습니다!

directory 구조도 다른 분들이랑 동일하게 기존의 /ml/opt/final~ 에서 /ml/opt/input/final~로 변경했습니다. final project 폴더 통째로 input 폴더에 옮기고, 파일 몇개 ctrl+f 전체 찾기한 다음 전체 바꾸기 하니까 금방하더라구요. 참고해주세요.

뭐가 제일 좋은지는 잘 모르겠고 아래 3개 정도 해보시면 될 것 같습니다.

python finetune.py python finetune.py --I_I2_loss --lambda_I_I2=1 python finetune.py --I_I2_loss --lambda_I_I2=1 --lwf_sky_not_limit

Dongwoo-Im commented 2 years ago

그리고 CLAHE.py에서 RTTS 데이터에 clip_limit=1 grid_size=64 로 새로 만들어주셔야합니다!

Dongwoo-Im commented 2 years ago

fix/PSD branch도 refactor/PSD branch에 merge하겠습니다.

boostcampaitech3 / final-project-level3-cv-17