nagejacob / SpatiallyAdaptiveSSID

Spatially Adaptive Self-Supervised Learning for Real-World Image Denoising (CVPR 2023)
GNU General Public License v3.0
80 stars 6 forks source link

Loss function of LAN stage #4

Closed PyoJunCode closed 1 year ago

PyoJunCode commented 1 year ago

In paper Eq(4), loss function of $\mathcal{L}_{LAN}$ is $(1-\alpha)||sg(\tilde{x}_1)-\tilde{x}_2||_1$. But in code, it just using l1 for loss.

            # alpha = generate_alpha(BNN)
            # self.loss = self.criteron(BNN.detach() * (1 - alpha), LAN * (1 - alpha))
            self.loss = self.criteron(BNN, LAN)

What loss should I use in LAN stage?

nagejacob commented 1 year ago

Either will be ok.

PyoJunCode commented 1 year ago

Thanks for fast reply, closing this issue