diff options
| author | skal <pascal.massimino@gmail.com> | 2026-02-10 22:55:28 +0100 |
|---|---|---|
| committer | skal <pascal.massimino@gmail.com> | 2026-02-10 22:55:28 +0100 |
| commit | edbc5fad0c258f2277e1d6b9d0ee9463be713bc9 (patch) | |
| tree | 4d3d6a76e2a0676463f4c429cef3e7c237364f7e /workspaces/main/shaders/cnn/cnn_layer.wgsl | |
| parent | 2adcf1bac1ec651861930eb2af00641eb23f6ef1 (diff) | |
chore: Update CNN architecture to 3×3×3 with new trained weights
Changed from 3×5×3 to 3×3×3 architecture for testing.
Changes:
- cnn_layer.wgsl: Use 3×3 conv for all layers
- cnn_weights_generated.wgsl: Regenerated weights
- image_style_processor.py: Made executable
handoff(Claude): CNN mismatch analysis complete, patch extraction added, docs updated
Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
Diffstat (limited to 'workspaces/main/shaders/cnn/cnn_layer.wgsl')
| -rw-r--r-- | workspaces/main/shaders/cnn/cnn_layer.wgsl | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/workspaces/main/shaders/cnn/cnn_layer.wgsl b/workspaces/main/shaders/cnn/cnn_layer.wgsl index d33a301..48bdcc6 100644 --- a/workspaces/main/shaders/cnn/cnn_layer.wgsl +++ b/workspaces/main/shaders/cnn/cnn_layer.wgsl @@ -8,7 +8,6 @@ #include "common_uniforms" #include "cnn_activation" #include "cnn_conv3x3" -#include "cnn_conv5x5" #include "cnn_weights_generated" struct CNNLayerParams { @@ -43,7 +42,7 @@ struct CNNLayerParams { result = cnn_tanh(result); } else if (params.layer_index == 1) { - result = cnn_conv5x5_7to4(txt, smplr, uv, uniforms.resolution, + result = cnn_conv3x3_7to4(txt, smplr, uv, uniforms.resolution, gray, weights_layer1); result = cnn_tanh(result); // Keep in [-1,1] } |
