fix: parameter tuning and cleanup
This commit is contained in:
+7
-10
@@ -81,12 +81,12 @@ class VideoTAADLAA:
|
||||
return {
|
||||
"required": {
|
||||
"images": ("IMAGE",),
|
||||
"taa_strength": ("FLOAT", {"default": 0.60, "min": 0, "max": 1, "step": 0.05}),
|
||||
"taa_alpha": ("FLOAT", {"default": 0.40, "min": 0, "max": 0.9, "step": 0.01}),
|
||||
"motion_sensitivity": ("FLOAT", {"default": 0.08, "min": 0.0, "max": 0.3, "step": 0.01}),
|
||||
"taa_strength": ("FLOAT", {"default": 0.40, "min": 0, "max": 1, "step": 0.05}),
|
||||
"taa_alpha": ("FLOAT", {"default": 0.30, "min": 0, "max": 0.9, "step": 0.01}),
|
||||
"motion_sensitivity": ("FLOAT", {"default": 0.05, "min": 0.0, "max": 0.3, "step": 0.01}),
|
||||
"jitter_scale": ("FLOAT", {"default": 0.02, "min": 0, "max": 0.08, "step": 0.01}),
|
||||
"dlaa_strength": ("FLOAT", {"default": 0.30, "min": 0, "max": 1, "step": 0.01}),
|
||||
"edge_threshold": ("FLOAT", {"default": 0.20, "min": 0.05, "max": 0.35, "step": 0.01}),
|
||||
"dlaa_strength": ("FLOAT", {"default": 0.40, "min": 0, "max": 1, "step": 0.05}),
|
||||
"edge_threshold": ("FLOAT", {"default": 0.25, "min": 0.05, "max": 0.35, "step": 0.01}),
|
||||
"blur_radius": ("INT", {"default": 0, "min": 0, "max": 5, "step": 1}),
|
||||
"reset_history": ("BOOLEAN", {"default": False}),
|
||||
}
|
||||
@@ -172,12 +172,9 @@ class VideoTAADLAA:
|
||||
|
||||
# slight contrast adjustment
|
||||
mean_luma = torch.mean(luma_orig, dim=(1,2,3), keepdim=True)
|
||||
rgb = (rgb - mean_luma) * 1.02 + mean_luma
|
||||
rgb = (rgb - mean_luma) * 1.05 + mean_luma
|
||||
|
||||
# tiny channel tweak
|
||||
# rgb[:,0:1] *= 0.995
|
||||
# rgb[:,2:3] *= 1.005
|
||||
|
||||
rgb = torch.lerp(luma_orig, rgb, 1.15)
|
||||
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