Training self-made data，IoU and Accuracy is 0

[A-Chas]

My own dataset category is 4, class_ The program can run when the weight is 4, class_ An error will be reported if the weight is 5. However, there will always be a category whose Accuracy and Iou are 0. Please tell me what code needs to be changed if you want to run your own dataset. If we can, we can add a friend to communicate: [email protected]