There is tons of pre-existing datasets with various classes.
For your use-case a subset of a bigger dataset or a mix of classes from different datasets is often helpful.
Introducing https://pypi.org/project/YoloDatasetCustomizer/
- Give a list of paths to “data.yaml” files and/or just a top folder where all your local datasets reside.
- Select the class names you want to have in your custom dataset.
Enjoy your custom dataset ![]()