Suggestion of model

[Samuel_Bachorik]

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.
Image

Mask

[Dwight_Foster]

You could look at detectron2 here or maybe unet here.