Predict output black images

[darvida]

Hello i have trained the model with 3 classes. The data.py classes looks like: ` BackGround = [0, 0, 0] road = [2, 2, 2]

COLOR_DICT = np.array([BackGround, road])

one = [1, 1, 1]

COLOR_DICT = np.array([BackGround,road,one]) ... self.image_color_mode = "rgb" self.label_color_mode = "rgb"

    self.flag_multi_class = flag_multi_class
    self.num_class = num_classes
    self.target_size = (512, 512)
    self.img_type = 'jpg'

`

I have annotated the images with background color (0,0,0), class 1 (1,1,1) and class 2 (2,2,2). When training it went up to 98% accuracy early in the training, which might indicate that something might be wrong.

Can anyone help me out ?