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

Are there recommended input shape for each model? #66

Closed
tucan9389 opened this issue May 23, 2020 · 2 comments
Closed

Are there recommended input shape for each model? #66

tucan9389 opened this issue May 23, 2020 · 2 comments
Assignees
Labels
question Further information is requested

Comments

@tucan9389
Copy link

Hi, I'm an iOS app developer who is interesting in ML.
Firstly, thank you for sharing your awesome vision modeling project.

I want to support imgclsmob's pre-trained models on my repositories which provide baseline code for inferencing various ML models on iOS devices.

I converted all imgclsmob's pre-trained pose estimation model to tflite with (1,224,224,3) input shape. But I think each model has an optimized input shape that was used on training.

Are there recommended input shape for each model?

Here is the issue what I'm working:
tucan9389/PoseEstimation-TFLiteSwift#59

Thank you.

@osmr osmr self-assigned this May 24, 2020
@osmr osmr added the question Further information is requested label May 24, 2020
@osmr
Copy link
Owner

osmr commented May 24, 2020

Nice work!
To answer your question, see the source code of each model, in particular, the default value of the in_size argument. In most cases it's (1, 256, 192, 3).

@tucan9389
Copy link
Author

  • 256: height
  • 192: width

Right?

Now I’m testing all pose models provided by this repo. When I’m done, I’ll share it.

Thank you!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested
Projects
None yet
Development

No branches or pull requests

2 participants