-
Notifications
You must be signed in to change notification settings - Fork 26
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
code changes required for image size 256. #8
Comments
@abhinav3 You need to change hard-coded numbers in model.py (defining network), train.py, and test.py (preprocessing data). Yes, it will be great if you do so. Thanks! |
@abhinav3 Do you have an open PR for this? |
@woozzu What do you suggest should be an ideal way to take images of size 256/224? Should we just reshape the image to 128 or is the code generalized enough to take arbitrary image size if we just remove the hard coding? |
@shubhamagarwal92 You can simply change the hard-coded numbers. Everything works fine, but you may want to add more residual blocks or down/upsampling to enlarge a receptive field. |
Hi,
I wanted to run the code for images of size 256 as input instead of 128.
I see that image size 128 is hardcoded in "train.py" while transforming into tensors.
Is there any other place across the code where I need to change the image size (or other parameters dependent on image size) to run it on 256 size input image.
Should I raise a PR for the same where I can ask for the image size as run time argument?
Thanks, Abhinav
The text was updated successfully, but these errors were encountered: