Can this code support road mask as well?

I would like to get person mask + keypoint + road mask + sidewalk mask, so can i train the model for the road and sidewalk which doesn't have keypoint?
Please help