Added modelUrl support to tensorflow plugin.

2020-07-06 17:56:51 -04:00 · 2020-07-06 17:56:51 -04:00 · 81ac3bd1ea
parent f178d2121c
commit 81ac3bd1ea
1 changed files with 2 additions and 1 deletions
--- a/plugins/tensorflow/ObjectDetectors.js
+++ b/plugins/tensorflow/ObjectDetectors.js
@ -24,7 +24,8 @@ module.exports = function(config){

  async function loadCocoSsdModal() {
      const modal = await cocossd.load({
-          base: config.cocoBase || 'lite_mobilenet_v2' //lite_mobilenet_v2
+          base: config.cocoBase || 'lite_mobilenet_v2', //lite_mobilenet_v2
+          modelUrl: config.cocoUrl,
      })
      return modal;
  }