SemTorch: A Semantic Segmentation library build above FastAI

Hi, have anyone experiment with running semantic segmentation prediction with various input size and get the correct mask for it.
I trained deeplab3 model with fixed size of (75, 1075)
But i want to use it to predict a lot of different size inputs,
For example: (250, 4500), (350, 4500), …
The prediction results alway have size (75,1075).
I know in fastaiv1 we can do
learn.data.single_ds.tfmargs['size'] = None
Ref: Segmentation mask prediction on different input image sizes

But i dont know a good way to deal with this in fastaiv2.
Thanks

1 Like