burn/crates/burn-dataset/tests/data
Sergey 'Jin' Bostandzhyan a882843294
Extend ImageFolderDataset to support import of COCO detection (#2612)
COCO is a popular dataset which defines an own dataset format:
https://cocodataset.org/#format-data

This commit introduces a ImageFolderDataset::new_coco_detection()
function, which expects a path to the JSON annotations file in COCO
format and a path where the actual images are stored.

Note, that while COCO also offers segmentation and pose estimation data,
for now only the import of detection (bounding boxes) data is supported.
2024-12-20 08:04:54 -05:00
..
image_folder [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
image_folder_coco Extend ImageFolderDataset to support import of COCO detection (#2612) 2024-12-20 08:04:54 -05:00
segmask_folder Add segmentation mask to ImageFolderDataset (#2426) 2024-11-11 12:23:31 -05:00
dataset-fmt.csv [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
dataset.csv [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
dataset.json [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
dataset_coco.json Extend ImageFolderDataset to support import of COCO detection (#2612) 2024-12-20 08:04:54 -05:00
sqlite-dataset.db [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00