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
Hi, I want to thank you for your great work.
Recently, I've used YOLOv5-obb for training on my custom dataset with 2 bounding box classes. The issue is the inference's False Negative (FN) is too high, and I cannot improve the performance after many trials, I've tried to change some configurations, such as: learning rate, weight decay, even adding my own AdamW optimizer. My data is a combination of a private set and a public set called Retail50K (example below) and I cannot share here, but it is for groceries rows detection, and the objects (rows) bounding box labels often horizontal and long (many x coordinates is equal to 0 or image width), so I also remove the label mask in your dataset code to make sure no labels are removed:
I also used multi-scale training as well as your provided pretrained weights for yolov5m but the problem does not seemed to be solved. What could be the issue here? Here is an example of Retail50K dataset:
I used the image size of (640, 640) for training.
The text was updated successfully, but these errors were encountered:
Hi, I want to thank you for your great work.
Recently, I've used YOLOv5-obb for training on my custom dataset with 2 bounding box classes. The issue is the inference's False Negative (FN) is too high, and I cannot improve the performance after many trials, I've tried to change some configurations, such as: learning rate, weight decay, even adding my own AdamW optimizer. My data is a combination of a private set and a public set called Retail50K (example below) and I cannot share here, but it is for groceries rows detection, and the objects (rows) bounding box labels often horizontal and long (many x coordinates is equal to 0 or image width), so I also remove the label mask in your dataset code to make sure no labels are removed:
I also used multi-scale training as well as your provided pretrained weights for yolov5m but the problem does not seemed to be solved. What could be the issue here? Here is an example of Retail50K dataset:
I used the image size of (640, 640) for training.
The text was updated successfully, but these errors were encountered: