Tool that makes it easy to split YOLOs images and their associated labels into separate sets for training and testing.
Project description
yolo-splitter
Tool that makes it easy to split YOLOs images and their associated labels into separate sets for training and testing.
Installation
pip install yolosplitter
Uses
from yolosplitter import YoloSplitter
# give directory path containing Image and Labels
ys = YoloSplitter(input_dir="MyDataset/")
# creates the dataframe
df = ys.create_dataframe()
# saves the Images and labels in "new_dataset" dir. with data.yaml
ys.split_and_save_project(DF=df,output_dir="MyDataset-splitted",train_size=0.70)
# Dataframe contains Image names, Label names, annoations and class names.
# In the dataframe, we can observe the number of classes present in each image.
Input Directory
MyDataset/
├── 02.png
├── 02.txt
├── 03.png
├── 03.txt
├── 04.png
├── 04.txt
├── 05.png
├── 05.txt
├── 06.png
├── 06.txt
├── 07.png
├── 07.txt
├── 08.png
├── 08.txt
├── 09.png
├── 09.txt
├── 10.png
├── 10.txt
├── 11.png
└── 11.txt
Output Directory
MyDataset-splitted/
├── data.yaml
├── train
│ ├── images
│ │ ├── 03.png
│ │ ├── 04.png
│ │ ├── 05.png
│ │ ├── 07.png
│ │ ├── 08.png
│ │ ├── 09.png
│ │ └── 10.png
│ └── labels
│ ├── 03.txt
│ ├── 04.txt
│ ├── 05.txt
│ ├── 07.txt
│ ├── 08.txt
│ ├── 09.txt
│ └── 10.txt
└── val
├── images
│ ├── 02.png
│ ├── 06.png
│ └── 11.png
└── labels
├── 02.txt
├── 06.txt
└── 11.txt
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
yolosplitter-0.0.1.tar.gz
(3.4 kB
view hashes)
Built Distribution
Close
Hashes for yolosplitter-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2061a168c8516c430f8be4f8552c8dc34ed135740756d14e3ea321867294f390 |
|
MD5 | 4305f49530e56acfaaf5ea47ced7531d |
|
BLAKE2b-256 | 6a70f495f8fe8a138707b5d095735dc2898ba7c08751a4127a1b3e01cc4b730c |