imgly/background-removal-js

Can you release model with smaller input size?

seedlingfl opened this issue · 2 comments

The input size 1024x1024 uses too much memory for my case, would you consider release models with smaller input size, such as 512x512?
Or directly convert your pretrained model (pytorch?) to onnx using 512x512 input size might also work well.

With the static onnx model, it seems very hard to change the input size.

Is the model size the problem?

Actually I am able to use the 1024x1024 model in my case now. Thanks.
But it will still be good if there is option to use smaller input size to reduce memory usage.