summaryrefslogtreecommitdiff
path: root/workspaces/main/shaders/cnn/cnn_layer.wgsl
diff options
context:
space:
mode:
authorskal <pascal.massimino@gmail.com>2026-02-11 23:25:09 +0100
committerskal <pascal.massimino@gmail.com>2026-02-11 23:25:09 +0100
commit75b820e1d5be15b0187bb201ca432157b4049bc5 (patch)
tree28f035e4647212ca518c9b2772ab1e565f26e9c0 /workspaces/main/shaders/cnn/cnn_layer.wgsl
parent8ff8c56cd68d9e785cf6cb36ce1fc2bdc54ac15a (diff)
docs: Update CNN comments and add bias fix summary
- Fix stale comments: RGBD→RGB (not grayscale) - Clarify shape transformations in inference - Add CNN_BIAS_FIX_2026-02.md consolidating recent fixes - Include regenerated weights with 5x5 kernel for layer 0 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.wgsl3
1 files changed, 2 insertions, 1 deletions
diff --git a/workspaces/main/shaders/cnn/cnn_layer.wgsl b/workspaces/main/shaders/cnn/cnn_layer.wgsl
index e629cb2..cbd1686 100644
--- a/workspaces/main/shaders/cnn/cnn_layer.wgsl
+++ b/workspaces/main/shaders/cnn/cnn_layer.wgsl
@@ -8,6 +8,7 @@
#include "common_uniforms"
#include "cnn_activation"
#include "cnn_conv3x3"
+#include "cnn_conv5x5"
#include "cnn_weights_generated"
struct CNNLayerParams {
@@ -37,7 +38,7 @@ struct CNNLayerParams {
// Layer 0: 7→4 (RGBD output, normalizes [0,1] input)
if (params.layer_index == 0) {
- result = cnn_conv3x3_7to4_src(txt, smplr, uv, uniforms.resolution, weights_layer0);
+ result = cnn_conv5x5_7to4_src(txt, smplr, uv, uniforms.resolution, weights_layer0);
result = cnn_tanh(result);
}
else if (params.layer_index == 1) {