Browse Source

Update view.js

Lutz Roeder 6 years ago
parent
commit
a3ae247ebc
3 changed files with 3 additions and 0 deletions
  1. 1 0
      Makefile
  2. 1 0
      setup.py
  3. 1 0
      src/view.js

+ 1 - 0
Makefile

@@ -30,6 +30,7 @@ update:
 	@./tools/darknet sync
 	@./tools/dl4j sync
 	@./tools/keras sync install metadata
+	@./tools/mediapipe sync
 	@./tools/mlnet sync metadata
 	@./tools/mnn sync schema
 	@./tools/mxnet sync metadata

+ 1 - 0
setup.py

@@ -100,6 +100,7 @@ setuptools.setup(
             'dl4j.js', 'dl4j-metadata.json',
             'flux.js', 'flux-metadata.json',
             'keras.js', 'keras-metadata.json',
+            'mediapipe.js',
             'mlnet.js', 'mlnet-metadata.json',
             'mnn.js', 'mnn-metadata.json', 'mnn-schema.js',
             'mxnet.js', 'mxnet-metadata.json',

+ 1 - 0
src/view.js

@@ -1177,6 +1177,7 @@ view.ModelFactoryService = class {
         this.register('./torch', [ '.t7' ]);
         this.register('./tflite', [ '.tflite', '.lite', '.tfl', '.bin' ]);
         this.register('./tf', [ '.pb', '.meta', '.pbtxt', '.prototxt', '.json', '.index', '.ckpt' ]);
+        this.register('./mediapipe', [ '.pbtxt' ]);
         this.register('./sklearn', [ '.pkl', '.joblib', '.model' ]);
         this.register('./cntk', [ '.model', '.cntk', '.cmf', '.dnn' ]);
         this.register('./paddle', [ '.paddle', '__model__' ]);