FrancescoSaverioZuppichini / glasses

High-quality Neural Networks for Computer Vision 😎
https://francescosaveriozuppichini.github.io/glasses-webapp/
MIT License
436 stars 37 forks source link

Is there a way to change the inputlayer to a custom size? #290

Closed Boraly closed 2 years ago

Boraly commented 2 years ago

Hello,

I would like to train a ResNet with my own dataset. This consists of HD images, so with the resolution 1280 x 720. Therefore the question, is it possible to change the input layer accordingly? And if so, how do I change it?

Thank you very much, Boraly

FrancescoSaverioZuppichini commented 2 years ago

Hi @Boraly ,

I hope you are doing great. Keep in mind that the size of the images is way too big, I advise resizing it to something less computation expensive. Saying that you can pass any resolution you want to the model, the only input constrain for the default ResNet is to have 3 channels :)

Let me know if it works for you :)

Cheers,

Francesco