Skip to content

A simple python script to split data into train, test, and validation sets.

Notifications You must be signed in to change notification settings

bigrobinson/Training-Data-Splitter

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 

Repository files navigation

Training-Data_Splitter

A simple python module to split image training data into training, test, and validation directories.

I have used this for training a YOLOv3 model. Data must be split into directories by class. The root directory shall contain one images directory for each class and one labels directory for each class. The name of the labels directory shall be the same as the image directory with '_labels' appended to the end, like this:

images directory name for a single class: class1

labels directory name for a single class: class1_labels

The data will be organized into directories called 'images' and 'labels' with subdirectories for train, test, and validation sets.

Sample execution:

python3 org_data.py

About

A simple python script to split data into train, test, and validation sets.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages