Merge branch 'tensorflow_modelUrl_support' into 'dev'

Added modelUrl support to tensorflow plugin. See merge request Shinobi-Systems/Shinobi!213
2020-07-07 21:02:57 +00:00 · 2020-07-07 21:02:57 +00:00 · 6fbd85ce51
parent f178d2121c 81ac3bd1ea
commit 6fbd85ce51
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;
  }