diff options
Diffstat (limited to 'workspaces/main/shaders/cnn/cnn_layer.wgsl')
| -rw-r--r-- | workspaces/main/shaders/cnn/cnn_layer.wgsl | 3 |
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 48bdcc6..d33a301 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 { @@ -42,7 +43,7 @@ struct CNNLayerParams { result = cnn_tanh(result); } else if (params.layer_index == 1) { - result = cnn_conv3x3_7to4(txt, smplr, uv, uniforms.resolution, + result = cnn_conv5x5_7to4(txt, smplr, uv, uniforms.resolution, gray, weights_layer1); result = cnn_tanh(result); // Keep in [-1,1] } |