facebookarchive / fb.resnet.torch

Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts
Other
2.29k stars 664 forks source link

Retrain with different input size #192

Open pconway11 opened 6 years ago

pconway11 commented 6 years ago

How can I retrain resnet with a different image size than 224 * 224

I have removed the cropping from my copy of the train and classify script.

I can train successfuly as long as I provide images which are 224x224

But I can't figure out where this size is specified or if it can be changed.

I have a particular dataset which means cropping is not an option, and resizing to 224x224 obscures some of the important image features