summaryrefslogtreecommitdiff
path: root/workspaces/main/shaders/cnn/cnn_conv7x7.wgsl
diff options
context:
space:
mode:
Diffstat (limited to 'workspaces/main/shaders/cnn/cnn_conv7x7.wgsl')
-rw-r--r--workspaces/main/shaders/cnn/cnn_conv7x7.wgsl53
1 files changed, 0 insertions, 53 deletions
diff --git a/workspaces/main/shaders/cnn/cnn_conv7x7.wgsl b/workspaces/main/shaders/cnn/cnn_conv7x7.wgsl
deleted file mode 100644
index e68d644..0000000
--- a/workspaces/main/shaders/cnn/cnn_conv7x7.wgsl
+++ /dev/null
@@ -1,53 +0,0 @@
-// 7x7 convolution with 49 samples
-// Applies mat4 weights per sample
-
-fn cnn_conv7x7(
- tex: texture_2d<f32>,
- samp: sampler,
- uv: vec2<f32>,
- resolution: vec2<f32>,
- weights: array<mat4x4<f32>, 49>,
- bias: vec4<f32>
-) -> vec4<f32> {
- let step = 1.0 / resolution;
- var sum = bias;
- var idx = 0;
-
- for (var dy = -3; dy <= 3; dy++) {
- for (var dx = -3; dx <= 3; dx++) {
- let offset = vec2<f32>(f32(dx), f32(dy)) * step;
- let sample = textureSample(tex, samp, uv + offset);
- sum += weights[idx] * sample;
- idx++;
- }
- }
-
- return sum;
-}
-
-fn cnn_conv7x7_with_coord(
- tex: texture_2d<f32>,
- samp: sampler,
- uv: vec2<f32>,
- resolution: vec2<f32>,
- rgba_weights: array<mat4x4<f32>, 49>,
- coord_weights: mat2x4<f32>,
- bias: vec4<f32>
-) -> vec4<f32> {
- let step = 1.0 / resolution;
- var sum = bias;
-
- sum += coord_weights * uv;
-
- var idx = 0;
- for (var dy = -3; dy <= 3; dy++) {
- for (var dx = -3; dx <= 3; dx++) {
- let offset = vec2<f32>(f32(dx), f32(dy)) * step;
- let rgba = textureSample(tex, samp, uv + offset);
- sum += rgba_weights[idx] * rgba;
- idx++;
- }
- }
-
- return sum;
-}