train on myown dataset

[hadi1994shokati]

Hello. Thank you so much for sharing your interesting work. I want to use this code for soil erosion binary segmentation. My segmentation task doesn't have base classes and just has 0 and 1 (erosion and non-erosion). Now I have two questions. 1: Is this code able to do this segmentation? 2: If yes, should I use just one split?
Thank you in advance.
Hadi

[JiaojiaoYe1994]

Hi, have you solved this problem? How do you splot your dataset?