立即学习:https://edu.csdn.net/course/play/26956/347461?utm_source=blogtoedu
各位,在用unet训练的时候报类别数目错误,用自己的数据,单个分类,在numclasses那里写的是2类,tf.keras.utils.to_categorical(cv2.imread(seg, 0),num_classes=self.num_classes)这句话报错IndexError: index 9 is out of bounds for axis 1 with size2,如何解决?