|
@@ -76,7 +76,7 @@ def detect_seal(image_np, model):
|
|
with sess1.as_default():
|
|
with sess1.as_default():
|
|
with sess1.graph.as_default():
|
|
with sess1.graph.as_default():
|
|
try:
|
|
try:
|
|
- out_boxes, out_scores, out_classes = model.predict([image_resize, image_shape])
|
|
|
|
|
|
+ out_boxes, out_scores, out_classes = model.predict([image_resize, image_shape],steps=1)
|
|
except ValueError as e:
|
|
except ValueError as e:
|
|
if '(0, 4)' in traceback.format_exc():
|
|
if '(0, 4)' in traceback.format_exc():
|
|
log("there is no seal!")
|
|
log("there is no seal!")
|