Li-Chongyi / Zero-DCE

Zero-DCE code and model
774 stars 187 forks source link

Loss functions Implementation #40

Open he-zhe opened 2 years ago

he-zhe commented 2 years ago

Hello,

For Color Constancy Loss and Illumination Smoothness Loss, the implementation seems a bit different from the equations published in the paper.

https://github.com/Li-Chongyi/Zero-DCE/blob/43f9da935d55758d93b54a47ce5872c96b342668/Zero-DCE_code/Myloss.py#L20-L23

There seems to me 3 power of 2 calls for each diff.

https://github.com/Li-Chongyi/Zero-DCE/blob/43f9da935d55758d93b54a47ce5872c96b342668/Zero-DCE_code/Myloss.py#L87 power of 2 is used instead of abs.