Skip to content
Snippets Groups Projects
Commit 053213a6 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent f97cd3fa
Branches
No related tags found
No related merge requests found
......@@ -530,7 +530,7 @@ class SRCNN:
conv_hkm = conv_hkm_b = tf.keras.layers.Conv2D(num_filters, kernel_size=KERNEL_SIZE, kernel_initializer='he_uniform', activation=activation, padding='VALID')(input_hkm_2d)
print(conv_hkm.shape)
num_filters_hkm = 32
num_filters_hkm = 8
conv_hkm = build_conv2d_block(conv_hkm, num_filters_hkm, activation, 'Conv2D_1')
num_filters_hkm *= 2
conv_hkm = build_conv2d_block(conv_hkm, num_filters_hkm, activation, 'Conv2D_2')
......@@ -549,7 +549,7 @@ class SRCNN:
conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_3', kernel_size=KERNEL_SIZE, scale=scale)
#conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_4', kernel_size=KERNEL_SIZE, scale=scale)
conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_4', kernel_size=KERNEL_SIZE, scale=scale)
#conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_5', kernel_size=KERNEL_SIZE, scale=scale)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment