isr/isr_interface.py

@@ -41,7 +41,7 @@ from isr.model import get_tiny_inference_model, seal_model, seal_model_se
 from isr.pre_process import count_red_pixel, get_anchors, get_classes, get_colors
 from isr.utils import get_best_predict_size, pil_resize, letterbox_image, draw_boxes, adjust_boxes
 from flask import Flask, request
-
+tf.compat.v1.disable_eager_execution()
 sess1 = tf.compat.v1.Session(graph=tf.Graph())
 sess2 = tf.compat.v1.Session(graph=tf.Graph())
 
@@ -100,7 +100,7 @@ def detect_seal(image_np, model):
     # inference data
     with sess1.as_default():
         with sess1.graph.as_default():
-            out_boxes, out_scores, out_classes = model.predict([image_resize, image_shape], steps=2)
+            out_boxes, out_scores, out_classes = model.predict([image_resize, image_shape], steps=1)
     if int(out_boxes.shape[0]) == 0:
         log("there is no seal!")
         return image_np, [], []