Use image type for CoreML input, reshape last layer to work with metal shader

2021-09-10 22:13:41 +09:30
parent a72f04006f
commit 28b11aaa44
2 changed files with 3 additions and 2 deletions
--- a/fast_depth_functional.py
+++ b/fast_depth_functional.py
@@ -72,6 +72,7 @@ def mobilenet_nnconv5(weights=None, shape=(224, 224, 3)):
    x = keras.layers.Conv2D(1, 1, padding='same')(x)
    x = keras.layers.BatchNormalization()(x)
    x = keras.layers.ReLU(6.)(x)
+    x = keras.layers.Reshape([shape[0], shape[1]])(x)
    return keras.Model(inputs=input, outputs=x, name="fast_depth")