summaryrefslogtreecommitdiff
path: root/res/TensorFlowPythonExamples/examples/non_max_suppression_padded_2/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'res/TensorFlowPythonExamples/examples/non_max_suppression_padded_2/__init__.py')
-rw-r--r--res/TensorFlowPythonExamples/examples/non_max_suppression_padded_2/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/res/TensorFlowPythonExamples/examples/non_max_suppression_padded_2/__init__.py b/res/TensorFlowPythonExamples/examples/non_max_suppression_padded_2/__init__.py
index 42e7bf06c..932ad3534 100644
--- a/res/TensorFlowPythonExamples/examples/non_max_suppression_padded_2/__init__.py
+++ b/res/TensorFlowPythonExamples/examples/non_max_suppression_padded_2/__init__.py
@@ -1,5 +1,7 @@
import tensorflow as tf
+tf.compat.v1.disable_eager_execution()
+
max_output_size = tf.compat.v1.constant(6)
iou_threshold = tf.compat.v1.constant(0.5)
score_threshold = tf.compat.v1.constant(0.6)