Set different height and weight for custom model #313
-
How to train a custom model with different height and weight? Let's say our input image size height is 1088 and weight is 4096, if I update custom model config file it only takes one parameter as input and send it as same height and weight to network if I am not mistaken. How do I update different sizes in config file for custom model? If I update my
|
Beta Was this translation helpful? Give feedback.
Replies: 1 comment
-
Hi, I assume you mean width instead of weight. In that case try to introduce the size in a list. Hope it works! |
Beta Was this translation helpful? Give feedback.
Hi, I assume you mean width instead of weight. In that case try to introduce the size in a list.
For example, in your case you have to type:
image_size: [1088, 4096]
Hope it works!