You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
First of all, thank you very much for sharing your paper and code. I have also read the previous answers regarding issues with training on custom datasets, but I still haven't resolved the problems I encountered during my own dataset training.
My dataset has two classes (0 for background, 1 for another category). I modified num_classes = 2 in voc.py and also adjusted the color mapping in palette.py with def get_voc_palette(num_classes) to represent two classes (black and white). I am training using a supervised approach, but I am encountering a situation where the intersection over union (IoU) for one of the classes is 0. Additionally, when loading the same model weights in inference.py, the output results vary with each run. I have no idea how to resolve these issues. Any guidance would be greatly appreciated
The text was updated successfully, but these errors were encountered:
First of all, thank you very much for sharing your paper and code. I have also read the previous answers regarding issues with training on custom datasets, but I still haven't resolved the problems I encountered during my own dataset training.
My dataset has two classes (0 for background, 1 for another category). I modified num_classes = 2 in voc.py and also adjusted the color mapping in palette.py with def get_voc_palette(num_classes) to represent two classes (black and white). I am training using a supervised approach, but I am encountering a situation where the intersection over union (IoU) for one of the classes is 0. Additionally, when loading the same model weights in inference.py, the output results vary with each run. I have no idea how to resolve these issues. Any guidance would be greatly appreciated
The text was updated successfully, but these errors were encountered: