Skip to content

Commit 74e8d2e

Browse files
committed
fix docs
1 parent f3af7d7 commit 74e8d2e

File tree

1 file changed

+159
-34
lines changed

1 file changed

+159
-34
lines changed

luxonis_ml/data/README.md

Lines changed: 159 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -387,53 +387,178 @@ for img, labels in loader:
387387
The supported formats are:
388388

389389
- **COCO** - We support COCO JSON format in two variants:
390+
391+
- [FiftyOne layout](https://docs.voxel51.com/user_guide/export_datasets.html#cocodetectiondataset-export)
392+
````plaintext
393+
dataset_dir/
394+
├── train/
395+
│ ├── data/
396+
│ │ ├── img1.jpg
397+
│ │ ├── img2.jpg
398+
│ │ └── ...
399+
│ └── labels.json
400+
├── validation/
401+
│ ├── data/
402+
│ └── labels.json
403+
└── test/
404+
├── data/
405+
└── labels.json
406+
```
407+
````
390408
- [RoboFlow](https://roboflow.com/formats/coco-json)
391-
- [FiftyOne](https://docs.voxel51.com/user_guide/export_datasets.html#cocodetectiondataset-export)
409+
````plaintext
410+
dataset_dir/
411+
├── train/
412+
│ ├── img1.jpg
413+
│ ├── img2.jpg
414+
│ └── ...
415+
│ └── _annotations.coco.json
416+
├── valid/
417+
└── test/
418+
```
419+
````
420+
392421
- [**Pascal VOC XML**](https://roboflow.com/formats/pascal-voc-xml)
422+
423+
```plaintext
424+
dataset_dir/
425+
├── train/
426+
│ ├── img1.jpg
427+
│ ├── img1.xml
428+
│ └── ...
429+
├── valid/
430+
└── test/
431+
```
432+
393433
- [**YOLO Darknet TXT**](https://roboflow.com/formats/yolo-darknet-txt)
434+
435+
```plaintext
436+
dataset_dir/
437+
├── train/
438+
│ ├── img1.jpg
439+
│ ├── img1.txt
440+
│ ├── ...
441+
│ └── _darknet.labels
442+
├── valid/
443+
└── test/
444+
```
445+
394446
- [**YOLOv4 PyTorch TXT**](https://roboflow.com/formats/yolov4-pytorch-txt)
447+
448+
```plaintext
449+
dataset_dir/
450+
├── train/
451+
│ ├── img1.jpg
452+
│ ├── img2.jpg
453+
│ ├── ...
454+
│ ├── _annotations.txt
455+
│ └── _classes.txt
456+
├── valid/
457+
└── test/
458+
```
459+
395460
- [**MT YOLOv6**](https://roboflow.com/formats/mt-yolov6)
461+
462+
```plaintext
463+
dataset_dir/
464+
├── images/
465+
│ ├── train/
466+
│ │ ├── img1.jpg
467+
│ │ ├── img2.jpg
468+
│ │ └── ...
469+
│ ├── valid/
470+
│ └── test/
471+
├── labels/
472+
│ ├── train/
473+
│ │ ├── img1.txt
474+
│ │ ├── img2.txt
475+
│ │ └── ...
476+
│ ├── valid/
477+
│ └── test/
478+
└── data.yaml
479+
```
480+
396481
- [**CreateML JSON**](https://roboflow.com/formats/createml-json)
482+
483+
```plaintext
484+
dataset_dir/
485+
├── train/
486+
│ ├── img1.jpg
487+
│ ├── img2.jpg
488+
│ └── ...
489+
│ └── _annotations.createml.json
490+
├── valid/
491+
└── test/
492+
```
493+
397494
- [**TensorFlow Object Detection CSV**](https://roboflow.com/formats/tensorflow-object-detection-csv)
495+
496+
```plaintext
497+
dataset_dir/
498+
├── train/
499+
│ ├── img1.jpg
500+
│ ├── img2.jpg
501+
│ ├── ...
502+
│ └── _annotations.csv
503+
├── valid/
504+
└── test/
505+
```
506+
398507
- [**SOLO**](https://docs.unity3d.com/Packages/[email protected]/manual/Schema/SoloSchema.html)
399-
- **Classification Directory** - A directory with subdirectories for each class
400508
401-
```plaintext
402-
dataset_dir/
403-
├── train/
404-
│ ├── class1/
405-
│ │ ├── img1.jpg
406-
│ │ ├── img2.jpg
407-
│ │ └── ...
408-
│ ├── class2/
409-
│ └── ...
410-
├── valid/
411-
└── test/
412-
```
509+
```plaintext
510+
dataset_dir/
511+
├── train/
512+
│ ├── metadata.json
513+
│ ├── sensor_definitions.json
514+
│ ├── annotation_definitions.json
515+
│ ├── metric_definitions.json
516+
│ └── sequence.<SequenceNUM>/
517+
│ ├── step<StepNUM>.camera.jpg
518+
│ ├── step<StepNUM>.frame_data.json
519+
│ └── (OPTIONAL: step<StepNUM>.camera.semantic segmentation.jpg)
520+
├── valid/
521+
└── test/
522+
```
413523
414-
- **Segmentation Mask Directory** - A directory with images and corresponding masks.
524+
- **Classification Directory** - A directory with subdirectories for each class
415525
416-
```plaintext
417-
dataset_dir/
418-
├── train/
419-
│ ├── img1.jpg
420-
│ ├── img1_mask.png
421-
│ ├── ...
422-
│ └── _classes.csv
423-
├── valid/
424-
└── test/
425-
```
526+
```plaintext
527+
dataset_dir/
528+
├── train/
529+
│ ├── class1/
530+
│ │ ├── img1.jpg
531+
│ │ ├── img2.jpg
532+
│ │ └── ...
533+
│ ├── class2/
534+
│ └── ...
535+
├── valid/
536+
└── test/
537+
```
426538
427-
The masks are stored as grayscale PNG images where each pixel value corresponds to a class.
428-
The mapping from pixel values to class is defined in the `_classes.csv` file.
539+
- **Segmentation Mask Directory** - A directory with images and corresponding masks.
429540
430-
```csv
431-
Pixel Value, Class
432-
0, background
433-
1, class1
434-
2, class2
435-
3, class3
436-
```
541+
```plaintext
542+
dataset_dir/
543+
├── train/
544+
│ ├── img1.jpg
545+
│ ├── img1_mask.png
546+
│ ├── ...
547+
│ └── _classes.csv
548+
├── valid/
549+
└── test/
550+
```
551+
552+
The masks are stored as grayscale PNG images where each pixel value corresponds to a class.
553+
The mapping from pixel values to class is defined in the `_classes.csv` file.
554+
555+
```csv
556+
Pixel Value, Class
557+
0, background
558+
1, class1
559+
2, class2
560+
3, class3
561+
```
437562
438563
### Dataset Creation
439564

0 commit comments

Comments
 (0)