fizyr / keras-maskrcnn

Keras implementation of MaskRCNN object detection.
Apache License 2.0
406 stars 131 forks source link

Can i disable resize image while training if my image size is less then 800*1333 #80

Open amardeepjaiman opened 5 years ago

amardeepjaiman commented 5 years ago

Hi,

Just wanted to know as in RetinaNet we can give any size input to network. Should we resize the images before feeding it to network to 800x1333 as mentioned in generator.py code. My images are already 500x500. Can i disable resizing the images otherwise my images will be resized to 800x1333.

hgaiser commented 5 years ago

The functionality to set the image-min-side and image-max-side got added to retinanet at some point. If you want, you could make a PR to do the same for keras-maskrcnn. You can then set the image-min-side and image-max-side to 500 and 500 respectively.