tensorspace-team / tensorspace

Neural network 3D visualization framework, build interactive and intuitive model in browsers, support pre-trained deep learning models from TensorFlow, Keras, TensorFlow.js
https://tensorspace.org
Apache License 2.0
5.07k stars 443 forks source link

Change tf loader API #216

Closed syt123450 closed 5 years ago

syt123450 commented 5 years ago

Previously, when loading a tensorflow frozen model in TensorSpace, it is required to provide two parameters: modelUrl and weightUrl. As TensorSpace uses tfjs's new loading API, weightUrl is no longer required, so new tensorflow load API will look like this temporary:

model.load( {
    type: "tensorflow",
    modelUrl: "/PATH_TO_MODEL/tensorflowjs_model.pb",
    outputsName: [ ... ],
} );

Next step is to change modelUrl to url, make tensorflow loader conform with keras loader and tfjs loader.

TODO: