search

xuebinqin / DIS

This is the repo for our new project Highly Accurate Dichotomous Image Segmentation
Apache License 2.0
2.27k stars 270 forks source link

image size for model #69

Open Zarxrax opened 1 year ago

Zarxrax commented 1 year ago

Hi, it looks like your code is optimized to train a model that works best for 1024x1024 resolution images. I am wondering if its fine to use other resolutions? Does it have to be square? For instance, if I intend a model to be used on HD video (1920x1080), would it be better for me to train at that specific size?

swframe commented 1 year ago

The code will resize your input image so the model is happy and it will resize the output back to the input size. It still works well.