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

Can I train ssd mobilenet with rectangular shape resize? #1886

Open
oguzhankayaa opened this issue Aug 16, 2024 · 1 comment
Open

Can I train ssd mobilenet with rectangular shape resize? #1886

oguzhankayaa opened this issue Aug 16, 2024 · 1 comment

Comments

@oguzhankayaa
Copy link

I want to train ssd mobilenet with rectangular shape image. As I understand train file is written only for square shape resizing. How can I resize in a rectangular form so that I can keep the aspect ratio?

@dusty-nv
Copy link
Owner

@oguzhankayaa I'm not sure, I'm not the original author of the pytorch-ssd repo - the upstream is from here: https://github.com/qfgaohao/pytorch-ssd

You can double check it in the code, but I believe it does center cropping so the aspect ratio is maintained. But you would loose the data on the edges. It seems more common to just train the model on squished aspect ratio if you don't want to modify the architecture.

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

No branches or pull requests

2 participants