update train_cnn.py and shader

1 files changed, 2 insertions, 11 deletions

diff --git a/training/train_cnn.py b/training/train_cnn.py
index 16f8e7a..902daa8 100755
--- a/training/train_cnn.py
+++ b/training/train_cnn.py
@@ -199,20 +199,11 @@ def generate_layer_shader(output_path, num_layers, kernel_sizes):
                 f.write(f"        let gray_out = {conv_fn}(txt, smplr, uv, uniforms.resolution,\n")
                 f.write(f"                                         original, weights_layer{layer_idx});\n")
                 f.write(f"        // gray_out already in [0,1] from clipped training\n")
-                f.write(f"        let original_denorm = (original + 1.0) * 0.5;\n")
                 f.write(f"        result = vec4<f32>(gray_out, gray_out, gray_out, 1.0);\n")
-                f.write(f"        let blended = mix(original_denorm, result, params.blend_amount);\n")
-                f.write(f"        return blended;  // [0,1]\n")
+                f.write(f"        return mix(original_raw, result, params.blend_amount); // [0,1]\n")
                 f.write(f"    }}\n")
 
-        # Add else clause for invalid layer index
-        if num_layers > 0:
-            f.write(f"    else {{\n")
-            f.write(f"        return textureSample(txt, smplr, uv);\n")
-            f.write(f"    }}\n")
-
-        f.write("\n    // Non-final layers: denormalize for display\n")
-        f.write("    return (result + 1.0) * 0.5;  // [-1,1] → [0,1]\n")
+        f.write("    return result;  // [-1,1]\n")
         f.write("}\n")