Search code examples
tensorflowtensorflow-datasetsdeeplearning4jdeeplab

Tensorflow mIOU and pixel accuracy bug?


Let's say I started training a tensorflow model from scratch with 1000 training steps. I get the following result at the completion of training. enter image description here enter image description here

Now, lets say I want to train for 2000 training steps from the previously saved checkpoint. I get the mIOU and pixel_accuracy = 1.0. enter image description here

I am using '''TensorFlow v1.13.1'''. How can I fix this bug or problem ?


Solution

  • The problem was with my data-set. I was assigning background_tag=1, class_one=2, class_two=3. After modifying ground truth images python PIL everything worked normally.