vato007/fast-depth-tf
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Piv
2021-09-10 22:13:41 +09:30
parent a72f04006f
commit 28b11aaa44
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
coreml.py
View File

@@ -1,8 +1,8 @@
import coremltools as ct import coremltools as ct
def convert_coreml(model_path, save_path='../mobilenet_nnconv5.mlmodel'): def convert_coreml(model_path, save_path='mobilenet_nnconv5.mlmodel'):
mlmodel = ct.convert(model_path) mlmodel = ct.convert(model_path, inputs=[ct.ImageType()])
mlmodel.save(save_path) mlmodel.save(save_path)

1
fast_depth_functional.py
View File

@@ -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.Conv2D(1, 1, padding='same')(x)
x = keras.layers.BatchNormalization()(x) x = keras.layers.BatchNormalization()(x)
x = keras.layers.ReLU(6.)(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") return keras.Model(inputs=input, outputs=x, name="fast_depth")