If you’re looking to experiment with a bunch of different types of segmentation models you can take a look at this repo: https://github.com/achaiah/torchsample/tree/frankenstein 266  I cloned @ncullen93’s excellent torchsample project and extended it into a bit of a monster  with a bunch of open-source segmentation and classification models provided.