zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Remove `check_requirements(('tensorflow>=2.4.1',))` (#5476) 

`check_requirements()` is unreliable for large packages like torch and tensorflow that may have multiple installation routes (i.e. conda, pip, tensorflow-cpu, etc.)
This commit is contained in:
Glenn Jocher 2021-11-03 14:51:37 +01:00 committed by GitHub
parent 042f02ff9b
commit 0eb37ad8af
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
detect.py
View File

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