masters-degree/longinus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Modify network arch

Browse Source
This commit is contained in:
Wojciech Janota 2024-08-02 14:27:06 +02:00
parent 23448a84e7
commit b2402f4336
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
longinus/longinus/coeff_encoders/neural_network.py
View File

@ -29,13 +29,13 @@ class NeuralNetworkEncoder:
self.model = keras.Sequential([
layers.Reshape((512, 512, 1), input_shape=(262144,)),
#layers.InputLayer(input_shape=(512 * 512, 1, 1)),
layers.Conv2D(32, (3, 3), activation=internal_activation_function, padding='same'),
layers.MaxPooling2D((2, 2)),
layers.Conv2D(64, (3, 3), activation=internal_activation_function, padding='same'),
layers.Conv2D(256, (3, 3), activation=internal_activation_function, padding='same'),
layers.MaxPooling2D((2, 2)),
layers.Conv2D(128, (3, 3), activation=internal_activation_function, padding='same'),
layers.MaxPooling2D((2, 2)),
layers.Conv2D(256, (3, 3), activation=internal_activation_function, padding='same'),
layers.Conv2D(64, (3, 3), activation=internal_activation_function, padding='same'),
layers.MaxPooling2D((2, 2)),
layers.Conv2D(32, (3, 3), activation=internal_activation_function, padding='same'),
layers.Flatten(),
layers.Dense(64, activation=external_activation_function)
])