Skip to content

Commit 8ed3de1

Browse files
committed
Addition to README
1 parent f46993a commit 8ed3de1

File tree

1 file changed

+61
-12
lines changed

1 file changed

+61
-12
lines changed

luxonis_ml/data/README.md

Lines changed: 61 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -557,19 +557,68 @@ The supported formats are:
557557
└── test/
558558
```
559559

560-
- **Classification Directory** - A directory with subdirectories for each class
560+
- **Classification Directory** - A directory with subdirectories for each class. Two structures are supported:
561561

562-
```plaintext
563-
dataset_dir/
564-
├── train/
565-
│ ├── class1/
566-
│ │ ├── img1.jpg
567-
│ │ ├── img2.jpg
568-
│ │ └── ...
569-
│ ├── class2/
570-
│ └── ...
571-
├── valid/
572-
└── test/
562+
- Split structure with train/valid/test subdirectories:
563+
```plaintext
564+
dataset_dir/
565+
├── train/
566+
│ ├── class1/
567+
│ │ ├── img1.jpg
568+
│ │ ├── img2.jpg
569+
│ │ └── ...
570+
│ ├── class2/
571+
│ └── ...
572+
├── valid/
573+
└── test/
574+
```
575+
- Flat structure (class subdirectories directly in root, random splits applied at parse time):
576+
```plaintext
577+
dataset_dir/
578+
├── class1/
579+
│ ├── img1.jpg
580+
│ └── ...
581+
├── class2/
582+
│ └── ...
583+
└── info.json (optional metadata file)
584+
```
585+
586+
- [**FiftyOne Classification**](https://docs.voxel51.com/user_guide/export_datasets.html#fiftyone-image-classification-dataset) - FiftyOneImageClassificationDataset format with images in a `data/` folder and labels in `labels.json`. Two structures are supported:
587+
588+
- Split structure with train/validation/test subdirectories:
589+
```plaintext
590+
dataset_dir/
591+
├── train/
592+
│ ├── data/
593+
│ │ ├── img1.jpg
594+
│ │ └── ...
595+
│ └── labels.json
596+
├── validation/
597+
│ ├── data/
598+
│ └── labels.json
599+
└── test/
600+
├── data/
601+
└── labels.json
602+
```
603+
- Flat structure (random splits applied at parse time):
604+
```plaintext
605+
dataset_dir/
606+
├── data/
607+
│ ├── img1.jpg
608+
│ └── ...
609+
└── labels.json
610+
```
611+
612+
The `labels.json` format:
613+
614+
```json
615+
{
616+
"classes": ["class1", "class2", ...],
617+
"labels": {
618+
"image_stem": class_index,
619+
...
620+
}
621+
}
573622
```
574623

575624
- **Segmentation Mask Directory** - A directory with images and corresponding masks.

0 commit comments

Comments
 (0)