diff --git a/detect.py b/detect.py index c57edba67c6c..f9c7bac3fca2 100644 --- a/detect.py +++ b/detect.py @@ -94,7 +94,6 @@ def run(weights=ROOT / 'yolov5s.pt', # model.pt path(s) import onnxruntime session = onnxruntime.InferenceSession(w, None) else: # TensorFlow models - check_requirements(('tensorflow>=2.4.1',)) import tensorflow as tf if pb: # https://www.tensorflow.org/guide/migrate#a_graphpb_or_graphpbtxt def wrap_frozen_graph(gd, inputs, outputs):