Object Detection (Training): Preprocessing - Githubissues

RedefiningReality / Pokedex-App

Using AI, take a picture of an unknown Pokémon and learn about its attributes. Completed for IPRO 497 at Illinois Tech

GNU General Public License v3.0

1 stars 1 forks source link

Object Detection (Training): Preprocessing #25

Open RedefiningReality opened 1 year ago

RedefiningReality commented 1 year ago

Create scripts that will prepare the dataset for training the Yolo algorithm. Script functionality:

setup.py

[x] Create images and annotations directories
[x] Put image and annotation files in their respective directories

prepare_training.py

[x] Extract data from XML annotation files
[x] Convert data to Yolov5 format
[x] Store data in respective TXT files
[x] Split the dataset into training, validation, and testing images
[x] Move images to their respective directories