Fix DLAANet weight loading

2026-04-20 17:27:00 +03:00
parent b89ab8060a
commit 6977e973a8
1 changed files with 5 additions and 5 deletions
@@ -196,17 +196,17 @@ class VideoTAADLAA:
                    
                    # apply residual mostly to luminance to avoid color shifts
                    luma_res = 0.2126 * residual[:, 0:1] + 0.7152 * residual[:, 1:2] + 0.0722 * residual[:, 2:3]
-                    rgb = rgb + (luma_res * dlaa_strength * 25.0)
+                    rgb = rgb + (luma_res * dlaa_strength * 32.0)
                    
                    # slight gamma & contrast adjustment
                    mean_luma = torch.mean(luma_orig, dim=(1,2,3), keepdim=True)
-                    rgb = (rgb - mean_luma) * 1.25 + (mean_luma * 1.07)
-                    rgb = torch.pow(rgb, 0.90)
+                    rgb = (rgb - mean_luma) * 1.03 + (mean_luma * 1.22)
+                    rgb = torch.pow(rgb, 0.88)
                    
                    rgb = torch.clamp(rgb, 0.0, 1.0)
-
+                    
                out_list.append(rgb.permute(0,2,3,1).cpu())
-
+                
                if i % 10 == 0:
                    mm.soft_empty_cache()