@@ -387,53 +387,178 @@ for img, labels in loader:
387387The 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