Any good code for inference segmentation?

[jamesp]

One relevant note from another thread - you can solve the size issue as follows:

Segmentation mask prediction on different input image sizes

Thanks a lot @sgugger, change learn.data.single_ds.tfmargs['size'] = None to predict one image per time worked like a charm!