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

Train models on non square images #10

Open
vmerc opened this issue Oct 4, 2018 · 1 comment
Open

Train models on non square images #10

vmerc opened this issue Oct 4, 2018 · 1 comment

Comments

@vmerc
Copy link

vmerc commented Oct 4, 2018

I tried to adapt them to train them on my own dataset but my images are rectangle. I've got errors about the tensor shape with ssd300VGG16 and ssd300XceptionLite durting the fit_generator call.

I adapted the input shape constant with my own values (1224,370,3) and change the Xception call with xception=Xception(input_shape=(1224,370,3),include_top=False,weights=None) but i still got errors about the tensor shape.

Does anyone no how to modify the models and the train notebook to make the models work ?

@tanakataiki
Copy link
Owner

The image should be resized to 300 by 300
You need to make new neteork if you want to detect with custom size if it's necessary

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