Three different dataset types are available to use in DL Workbench. Your dataset does not need to contain images from databases providing these types, but it needs to adhere to the supported dataset formats.
To learn how to download and prepare datasets, refer to Download and Cut Datasets.
NOTE: For Pascal VOC and COCO datasets, only the Object-Detection task is supported.
ImageNet is a well-known dataset used to train classification models. It consists of an annotation file and images:
The annotation file is organized as follows:
Pascal VOC is a well-known dataset used to train object-detection and semantic-segmentation models. VOC datasets consist of several folders containing annotation files and image indices.
A VOC dataset archive is organized as follows:
COCO dataset is used for object detection, segmentation, person keypoints detection, stuff segmentation, and caption generation.
A COCO dataset is downloaded as two separate archives, but you have to create one archive based on them as described in the Download COCO Dataset section in Download and Cut Datasets. To upload a COCO dataset to the DL Workbench, make sure the archive contains the following files:
A .json
file with annotations is organized as follows: