Skip to content
This repository was archived by the owner on Nov 3, 2022. It is now read-only.

Conversation

@ezavarygin
Copy link
Contributor

@ezavarygin ezavarygin commented Apr 9, 2019

At the moment, the MobileNets models try to load the weights which for some input_shape do not exist (one example is illustrated in this issue). This PR is related to this one.

Suggested solution:
Weights for the default size of 224 are loaded if rows is different from cols or not in [128, 160, 192, 224] for MobileNet_v1.
Weights for the default size of 224 are loaded if rows is different from cols or not in [96, 128, 160, 192, 224] for MobileNet_v2.

@taehoonlee
Copy link
Contributor

Thank you for the PR, @ezavarygin. Could you please add one missing case: undefined?

`input_shape` is undefined or non-square, or `rows` is not in [128, 160, 192, 224].

@ezavarygin
Copy link
Contributor Author

Done.

@taehoonlee taehoonlee closed this Apr 15, 2019
@taehoonlee taehoonlee reopened this Apr 15, 2019
@taehoonlee taehoonlee merged commit dda4997 into keras-team:master Apr 15, 2019
@ezavarygin ezavarygin deleted the custom_image_size_in_mobilenets branch April 15, 2019 11:36
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants