From 5990760f1cf4e6c35e7ea36b0551f98ff17c1c2f Mon Sep 17 00:00:00 2001
From: Mathieu Beligon <mathieu@feedly.com>
Date: Tue, 10 Mar 2020 00:12:28 -0400
Subject: [PATCH] [robots] (demo) fix: stop using settings to get model name
demo (temporary)
---
robots-at-robots/research/demos/utils.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/robots-at-robots/research/demos/utils.py b/robots-at-robots/research/demos/utils.py
index 93969a5..6bcbcdb 100644
--- a/robots-at-robots/research/demos/utils.py
+++ b/robots-at-robots/research/demos/utils.py
@@ -1,9 +1,8 @@
import tensorflow as tf
from polystar.common.constants import MODELS_DIR
-from polystar.robots_at_robots.globals import settings
def load_tf_model():
- model = tf.saved_model.load(export_dir=str(MODELS_DIR / settings.MODEL_NAME / "saved_model"))
+ model = tf.saved_model.load(export_dir=str(MODELS_DIR / "robots/ssd_mobilenet_v2_coco_2018_03_29" / "saved_model"))
return model.signatures["serving_default"]
--
GitLab