Python-Scripts
Python-Scripts copied to clipboard
Remove File mismatch for YOLO
For training a yolo model we label the images. Each labeled image has its corresponding text file with same name as the image containing coordinated of th bounding box in the image. But during the process either some text files or images are missing which hinders the training making the model think that the object is not in the image resulting in low accuracy. This script deletes any image with out label or any text file without image. In short it cleans the data, a part of data preprocessing.
i have requested a PR regarding this