From 81ac3bd1ea16362fadee80674199bcc48cb61bb7 Mon Sep 17 00:00:00 2001 From: Ian Rubado Date: Mon, 6 Jul 2020 17:56:51 -0400 Subject: [PATCH] Added modelUrl support to tensorflow plugin. --- plugins/tensorflow/ObjectDetectors.js | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/plugins/tensorflow/ObjectDetectors.js b/plugins/tensorflow/ObjectDetectors.js index 36137971..0270f213 100644 --- 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; }