Closed Dongwoo-Im closed 2 years ago
Commit을 쉽게 보기 위한 PR입니다.
test.py 수정 완료 했고, backbone으로 FFA & MSBDN & Dehazeformer 사용 가능합니다. Dehazeformer의 경우 메모리 문제로, 긴 축의 길이가 2560으로 제한됩니다.
test.py에서 사용하는 python file은 전체 4개이며, backbone model 3개 + dataloader 1개 입니다. 시연 과정에서 이미지가 하나씩 들어오기 때문에 dataloader는 필요없을 것 같고, 안에 있는 코드만 이식하면 될 것 같습니다.
참고로 저는 python 3.8.13 cuda 10.2 torch 1.8.1 사용 중입니다. backbone 3개 모두 동일 환경에서 돌아가며, dehazeformer는 모르겠지만 python 3.8.5에서도 돌아갈 것 같습니다.
python 3.8.13
cuda 10.2
torch 1.8.1
python 3.8.5
What is this PR?
Commit을 쉽게 보기 위한 PR입니다.
Changes
test.py 수정 완료 했고, backbone으로 FFA & MSBDN & Dehazeformer 사용 가능합니다. Dehazeformer의 경우 메모리 문제로, 긴 축의 길이가 2560으로 제한됩니다.
test.py에서 사용하는 python file은 전체 4개이며, backbone model 3개 + dataloader 1개 입니다. 시연 과정에서 이미지가 하나씩 들어오기 때문에 dataloader는 필요없을 것 같고, 안에 있는 코드만 이식하면 될 것 같습니다.
참고로 저는
python 3.8.13cuda 10.2torch 1.8.1사용 중입니다. backbone 3개 모두 동일 환경에서 돌아가며, dehazeformer는 모르겠지만python 3.8.5에서도 돌아갈 것 같습니다.To reviewers