how can i use in my own dataset (grayscale)

[YangBai1109]

I have just come into contact with deep learning, may I ask how I should modify this code if I want to do the binary classification of grayscale images, such as the segmentation of lumps