burn/crates/burn-dataset/tests
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
..
data Extend ImageFolderDataset to support import of COCO detection (#2612) 2024-12-20 08:04:54 -05:00