Forcing v1 behavior

theoway · Jun 16, 2022 · eb30044 · eb30044
1 parent 97b824c
commit eb30044
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/generate_detections.py b/tools/generate_detections.py
@@ -5,6 +5,7 @@
 import numpy as np
 import cv2
 import tensorflow.compat.v1 as tf
+tf.disable_v2_behavior()
 
 
 def _run_in_batches(f, data_dict, out, batch_size):