Skip to content
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

Merged
merged 9 commits into from
Sep 19, 2023

Conversation

Louis-Dupont
Copy link
Contributor

@Louis-Dupont Louis-Dupont commented Aug 27, 2023

Issue raised: model.predict implicitely expected (H, W, C) images.

Copy link
Contributor

@BloodAxe BloodAxe left a 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.

@shaydeci
Copy link
Contributor

don;'t forget to PR master320 branch too

Copy link
Contributor

@shaydeci shaydeci left a 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.

Copy link
Contributor

@shaydeci shaydeci left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@BloodAxe BloodAxe merged commit 4536d2d into master Sep 19, 2023
3 checks passed
@BloodAxe BloodAxe deleted the hotfix/SG-000-fix_predict_loading_from_np_torch branch September 19, 2023 11:25
BloodAxe pushed a commit that referenced this pull request Sep 19, 2023
* 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)
BloodAxe added a commit that referenced this pull request Sep 19, 2023
* 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]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants