1

2023-05-23 15:52:09 +08:00
parent e8f9c8a287
commit 663edbb5e2
88 changed files with 6250 additions and 0 deletions
@@ -0,0 +1,31 @@
+from __future__ import absolute_import
+from keras.layers.core import Dense, Activation, Flatten
+from keras.layers.convolutional import Conv2D, ZeroPadding2D
+
+
+def layers(input_shape):
+    return [
+        ZeroPadding2D((2, 2), input_shape=input_shape, data_format='channels_first'),
+        Conv2D(64, (5, 5), padding='valid', data_format='channels_first'),
+        Activation('relu'),
+
+        ZeroPadding2D((2, 2), data_format='channels_first'),
+        Conv2D(64, (5, 5), data_format='channels_first'),
+        Activation('relu'),
+
+        ZeroPadding2D((1, 1), data_format='channels_first'),
+        Conv2D(64, (3, 3), data_format='channels_first'),
+        Activation('relu'),
+
+        ZeroPadding2D((1, 1), data_format='channels_first'),
+        Conv2D(64, (3, 3), data_format='channels_first'),
+        Activation('relu'),
+
+        ZeroPadding2D((1, 1), data_format='channels_first'),
+        Conv2D(64, (3, 3), data_format='channels_first'),
+        Activation('relu'),
+
+        Flatten(),
+        Dense(512),
+        Activation('relu'),
+    ]