Balance Classes During YOLO Training Using a Weighted Dataloader
Implement class balancing in Ultralytics using a weighted dataloader and improve the performance of minority class.
I develop computer vision solutions for video analytics at the edge.
Implement class balancing in Ultralytics using a weighted dataloader and improve the performance of minority class. Get over 10% more mAP in small object detection by exploiting YOLOv8 pose models while training. Add additional classes to pre-trained YOLOv8 model without affecting the confidences on the existing classes. Some useful tricks and hacks I learnt while using MMDetection for my research work. A quick monkey patching technique to modify MMDetection 3.1.0 to return class probabilities.Recent Posts
Balance Classes During YOLO Training Using a Weighted Dataloader
A Simple Trick To Increase YOLOv8's Accuracy On Small Objects With No Overhead
Extending YOLOv8 COCO Model With New Classes Without Affecting Old Weights
MMDetection 3.x Hacks and Tricks
Monkey Patching MMDetection 3.1.0 To Get Class Probabilities
My Projects
View all »