Dear author, I am currently following this great work. May I ask which model you are using to segment an image into different parts.