--- title: StampDetection emoji: 🦀 colorFrom: gray colorTo: indigo sdk: gradio sdk_version: 4.31.0 app_file: app.py pinned: false license: mit --- # Stamp Detection using YoloV9 This repository shows how to custom train your model using YOLOV9. The script above uses a football players dataset to implement the process of custom training due to confidentiality with respect to the stamp dataset (PII information). However the same script can be used traing using custom dataset. ## Steps 1.) Firstly we will install labelme using python pip. Labelme is a graphical image annotation tool * pip install labelme 2.) Load your image directory in labelme and get to annotating 3.) Now once annotation is complete, install labelme2yolo to convert your labelme annotation to yolov9 format * labelme2yolo --json_dir /path/to/labelme_json_dir/ --val_size 0.15 --test_size 0.15 4.) Once done you can execute above script by replacing dataset with your custom path ## Training Performance The model was trained for 24 epochs on the stamp dataset. The training logs show the following metrics: Screenshot 2024-05-17 at 8 56 30 PM As you can see from above training logs: * Precision is 0.85 * Recall is 0.67 * mAp is 0.76 Screenshot 2024-05-12 at 1 49 36 PM Check out the github repo to know more - https://github.com/Himank-J/ERAV2/tree/main/S15