search

liuh127 / NTIRE-2021-Dehazing-Two-branch

Official PyTorch implementation of Two-branch Dehazing, wining runner-up awards in NTIRE 2021.
MIT License
51 stars 7 forks source link

MSSIM loss function #3

Closed Dpw506 closed 3 years ago

Dpw506 commented 3 years ago

Hi, Thank you for this work. I ran into the following problems while running the code you provided that is the MSSIM loss function you provided missing 1-MSSIM? What I see in the code is -MSSIM. How should I understand this? Looking forward to your reply.

liuh127 commented 3 years ago

Hi, Thank you for this work. I ran into the following problems while running the code you provided that is the MSSIM loss function you provided missing 1-MSSIM? What I see in the code is -MSSIM. How should I understand this? Looking forward to your reply.

Hi， please see the link for detailed implementation of MS-SSIM loss. https://github.com/liuh127/Two-branch-dehazing/blob/7fa08e810dba73088081384ec4a1940c872784e2/pytorch_msssim/__init__.py#L124

We use the implementation in https://github.com/jorge-pessoa/pytorch-msssim.

Dpw506 commented 3 years ago

Thank you very much!