Suggestion of model

Hello I want to ask if you can suggest me some good model for road segmentation. I have dataset with 2000 images from city and my small models just cant do this job for me. They have small capacity.
I leave one photo and mask from dataset.


You could look at detectron2 here or maybe unet here.

1 Like