Which YOLO version have you implemented? I’d advise to start out from pretrained weights to reduce overfitting. You can check if it overfits by comparing your train and val/test results. Does it work reeeeally good for train but not so good for val/test? That’s overfitting.