Heirudy
Heirudy
Thanks for your answer.I only trained with the first five images of the FIVES dataset and only changed the num_classes in your code, the two images below are the output...
#6 Since reading the previous discussion, I changed the num_classes to 255 and still reported the same error, so I changed it to 256 and it ran successfully after that.
My question is solved, thank you very much for your patience.
According to your suggestion, I changed the num_class to 2, and after the parameter divide was changed to True, the number of color channels pred of the model's prediction result...
My problem was successfully solved.This is the first deep learning program I have completed, and your patience and help have been invaluable.Words cannot fully express my gratitude and excitement. Thank...