We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Export
Based on this return statement in TFModel, the output of a TF saved_model was/should be a tensor. https://github.com/ultralytics/yolov5/blob/master/models/tf.py#L368
TFModel
However, after the changes introduced in #6292, the TF saved_model returns the said tensor wrapped in a list.
keras=True
keras=False
YOLOv5 Latest OS: Ubuntu, Windows Python: 3.9 TensorFlow: 2.7
No response
#7032 attempts to solve all these issues.
The text was updated successfully, but these errors were encountered:
@MrinalJain17 thanks for the PR! I will take a look there.
Sorry, something went wrong.
Successfully merging a pull request may close this issue.
Search before asking
YOLOv5 Component
Export
Bug
Based on this return statement in
TFModel
, the output of a TF saved_model was/should be a tensor. https://github.com/ultralytics/yolov5/blob/master/models/tf.py#L368However, after the changes introduced in #6292, the TF saved_model returns the said tensor wrapped in a list.
keras=True
) or the tensor wrapped in a listkeras=False
(now default).Environment
YOLOv5 Latest
OS: Ubuntu, Windows
Python: 3.9
TensorFlow: 2.7
Minimal Reproducible Example
No response
Additional
#7032 attempts to solve all these issues.
Are you willing to submit a PR?
The text was updated successfully, but these errors were encountered: