-
Notifications
You must be signed in to change notification settings - Fork 517
New issue
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
Hotfix/sg 000 fix predict loading from np torch #1419
Hotfix/sg 000 fix predict loading from np torch #1419
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This PR seems to be extending scope of supported input types that predict function supports. However no documentation reflecting this pretty serious change is updated. I think it must be explicitly stated in docs what is the expected (or supported) dtypes for predict() function can be and what will happen if one pass HWC/CHW/BHWC/BCHW inputs there.
Also see my inline comment for load_image
.
don;'t forget to PR master320 branch too |
…load_image method to get rid of hard-coded number of input channels
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
A few minor comments.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
* fix * Added table with all supported input types to predict() and improved load_image method to get rid of hard-coded number of input channels * Improve spelling * Improve type alias --------- Co-authored-by: Shay Aharon <[email protected]> Co-authored-by: Eugene Khvedchenya <[email protected]> (cherry picked from commit 4536d2d)
* fix * Added table with all supported input types to predict() and improved load_image method to get rid of hard-coded number of input channels * Improve spelling * Improve type alias --------- Co-authored-by: Shay Aharon <[email protected]> Co-authored-by: Eugene Khvedchenya <[email protected]> (cherry picked from commit 4536d2d) Co-authored-by: Louis-Dupont <[email protected]>
Issue raised:
model.predict
implicitely expected (H, W, C) images.