feat: DLAANet with residual learning and orthogonal weights
This commit is contained in:
+3
-1
@@ -85,7 +85,7 @@ class VideoTAADLAA:
|
||||
"taa_alpha": ("FLOAT", {"default": 0.40, "min": 0, "max": 0.9, "step": 0.01}),
|
||||
"motion_sensitivity": ("FLOAT", {"default": 0.05, "min": 0.01, "max": 0.5, "step": 0.01}),
|
||||
"jitter_scale": ("FLOAT", {"default": 0.02, "min": 0, "max": 0.08, "step": 0.01}),
|
||||
"dlaa_strength": ("FLOAT", {"default": 0.35, "min": 0, "max": 1, "step": 0.05}),
|
||||
"dlaa_strength": ("FLOAT", {"default": 0.30, "min": 0, "max": 1, "step": 0.05}),
|
||||
"edge_threshold": ("FLOAT", {"default": 0.25, "min": 0, "max": 0.5, "step": 0.01}),
|
||||
"blur_radius": ("INT", {"default": 0, "min": 0, "max": 5, "step": 1}),
|
||||
"reset_history": ("BOOLEAN", {"default": False}),
|
||||
@@ -157,6 +157,8 @@ class VideoTAADLAA:
|
||||
mean_luma = torch.mean(luma, dim=(1, 2, 3), keepdim=True)
|
||||
|
||||
rgb = (rgb - mean_luma) * 1.02 + mean_luma
|
||||
rgb[:, 2:3, :, :] += 0.025
|
||||
rgb[:, 0:1, :, :] -= 0.010
|
||||
rgb = torch.clamp(rgb, 0.0, 1.0)
|
||||
|
||||
out_list.append(rgb.permute(0, 2, 3, 1).cpu())
|
||||
|
||||
Reference in New Issue
Block a user