Update

archs/DualDn_arch.py

@@ -1,3 +1,10 @@
+# -----------------------------------------------------------------------------------
+# [ECCV2024] DualDn: Dual-domain Denoising via Differentiable ISP 
+# [Homepage] https://openimaginglab.github.io/DualDn/
+# [Author] Originally Written by Ruikang Li, from MMLab, CUHK.
+# [License] Absolutely open-source and free to use, please cite our paper if possible. :)
+# -----------------------------------------------------------------------------------
+
 import torch
 import torch.nn as nn
 from archs.backbone.restormer import Restormer
@@ -269,4 +276,4 @@ def forward(self, raw, colormask, wb_matrix, rgb_xyz_matrix, ref='D65', gamma_ty
         srgb = self.backbone2(in_srgb)
         out_srgb = self.conv_out2(srgb) + skip_srgb
 
-        return out_raw, out_srgb
+        return out_raw, out_srgb