|
@@ -1219,11 +1219,11 @@ view.ModelFactoryService = class {
|
|
|
this._extensions = [];
|
|
this._extensions = [];
|
|
|
this.register('./onnx', [ '.onnx', '.pb', '.pbtxt', '.prototxt', '.model' ]);
|
|
this.register('./onnx', [ '.onnx', '.pb', '.pbtxt', '.prototxt', '.model' ]);
|
|
|
this.register('./mxnet', [ '.mar', '.model', '.json', '.params' ]);
|
|
this.register('./mxnet', [ '.mar', '.model', '.json', '.params' ]);
|
|
|
|
|
+ this.register('./pytorch', [ '.pt', '.pth', '.pt1', '.pkl', '.h5', '.t7', '.model', '.dms', '.tar', '.ckpt', '.chkpt', '.bin', '.pb', '.zip' ]);
|
|
|
this.register('./keras', [ '.h5', '.hd5', '.hdf5', '.keras', '.json', '.cfg', '.model', '.pb', '.pth' ]);
|
|
this.register('./keras', [ '.h5', '.hd5', '.hdf5', '.keras', '.json', '.cfg', '.model', '.pb', '.pth' ]);
|
|
|
this.register('./coreml', [ '.mlmodel' ]);
|
|
this.register('./coreml', [ '.mlmodel' ]);
|
|
|
this.register('./caffe', [ '.caffemodel', '.pbtxt', '.prototxt', '.pt' ]);
|
|
this.register('./caffe', [ '.caffemodel', '.pbtxt', '.prototxt', '.pt' ]);
|
|
|
this.register('./caffe2', [ '.pb', '.pbtxt', '.prototxt' ]);
|
|
this.register('./caffe2', [ '.pb', '.pbtxt', '.prototxt' ]);
|
|
|
- this.register('./pytorch', [ '.pt', '.pth', '.pt1', '.pkl', '.h5', '.t7', '.model', '.dms', '.tar', '.ckpt', '.chkpt', '.bin', '.pb', '.zip' ]);
|
|
|
|
|
this.register('./torch', [ '.t7' ]);
|
|
this.register('./torch', [ '.t7' ]);
|
|
|
this.register('./tflite', [ '.tflite', '.lite', '.tfl', '.bin', '.pb', '.tmfile', '.h5', '.model', '.json' ]);
|
|
this.register('./tflite', [ '.tflite', '.lite', '.tfl', '.bin', '.pb', '.tmfile', '.h5', '.model', '.json' ]);
|
|
|
this.register('./tf', [ '.pb', '.meta', '.pbtxt', '.prototxt', '.pt', '.json', '.index', '.ckpt', '.graphdef', '.data-00000-of-00001' ]);
|
|
this.register('./tf', [ '.pb', '.meta', '.pbtxt', '.prototxt', '.pt', '.json', '.index', '.ckpt', '.graphdef', '.data-00000-of-00001' ]);
|