Remove `check_requirements(('tensorflow>=2.4.1',))` (#5476)
Browse files`check_requirements()` is unreliable for large packages like torch and tensorflow that may have multiple installation routes (i.e. conda, pip, tensorflow-cpu, etc.)
detect.py
CHANGED
|
@@ -94,7 +94,6 @@ def run(weights=ROOT / 'yolov5s.pt', # model.pt path(s)
|
|
| 94 |
import onnxruntime
|
| 95 |
session = onnxruntime.InferenceSession(w, None)
|
| 96 |
else: # TensorFlow models
|
| 97 |
-
check_requirements(('tensorflow>=2.4.1',))
|
| 98 |
import tensorflow as tf
|
| 99 |
if pb: # https://www.tensorflow.org/guide/migrate#a_graphpb_or_graphpbtxt
|
| 100 |
def wrap_frozen_graph(gd, inputs, outputs):
|
|
|
|
| 94 |
import onnxruntime
|
| 95 |
session = onnxruntime.InferenceSession(w, None)
|
| 96 |
else: # TensorFlow models
|
|
|
|
| 97 |
import tensorflow as tf
|
| 98 |
if pb: # https://www.tensorflow.org/guide/migrate#a_graphpb_or_graphpbtxt
|
| 99 |
def wrap_frozen_graph(gd, inputs, outputs):
|