edgeyolo icon indicating copy to clipboard operation
edgeyolo copied to clipboard

Minimum size for training

Open ramonhollands opened this issue 2 years ago • 2 comments

Hi Shihan Liu,

What is the reason there is a minimum input size for the model? In my experiments 192x192.

What would be needed to adapt the model to handle smaller sizes, eg 96x96?

Best regards, Ramon

ramonhollands avatar Jun 02 '23 19:06 ramonhollands

see params/train/train_settings.yaml

input_size: [672, 672]   # image input size for model
multiscale_range: 5      # real_input_size = input_size + randint(-multiscale_range, multiscale_range) * 32

so if you want to use a very small size, just set multiscale_range to zero

LSH9832 avatar Dec 06 '23 05:12 LSH9832

Thanks for your answer!

I fixed it by preventing random_resize (trainer.py) on small inputs, but this is the improved way.

verrassendhollands avatar Apr 22 '24 09:04 verrassendhollands