Просмотр исходного кода

Add experimental CoreML protobuf generator

Lutz Roeder 8 лет назад
Родитель
Сommit
1f02562318
1 измененных файлов с 22 добавлено и 0 удалено
  1. 22 0
      tools/coreml-generate

+ 22 - 0
tools/coreml-generate

@@ -0,0 +1,22 @@
+#!/bin/bash
+
+mkdir -p ../third_party
+
[email protected]:apple/coremltools.git
+
+if [ -d "../third_party/coremltools" ]; then
+    pushd "../third_party/coremltools" > /dev/null
+    echo "Fetch ${repository}..."
+    git fetch -p
+    echo "Reset ${repository}..."
+    git reset --hard origin/master
+    popd > /dev/null
+else
+    echo "Clone ${repository}..."
+    pushd "../third_party" > /dev/null
+    git clone --recursive ${repository}
+    popd > /dev/null
+fi
+
+echo "Generate '../src/coreml.js'"
+../node_modules/protobufjs/bin/pbjs -t static-module -w closure -r onnx -o ../src/coreml.js ../third_party/coremltools/mlmodel/format/Model.proto