Unverified 提交 db06f495 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

AutoUpdate TensorFlow in export.py (#9447)

上级 fe10b4ab
...@@ -309,6 +309,7 @@ def export_saved_model(model, ...@@ -309,6 +309,7 @@ def export_saved_model(model,
keras=False, keras=False,
prefix=colorstr('TensorFlow SavedModel:')): prefix=colorstr('TensorFlow SavedModel:')):
# YOLOv5 TensorFlow SavedModel export # YOLOv5 TensorFlow SavedModel export
check_requirements('tensorflow' if torch.cuda.is_available() else 'tensorflow-cpu')
import tensorflow as tf import tensorflow as tf
from tensorflow.python.framework.convert_to_constants import convert_variables_to_constants_v2 from tensorflow.python.framework.convert_to_constants import convert_variables_to_constants_v2
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论