Package org.jetbrains.kotlinx.dl.impl.dataset

Types

Coco

common

This enum represents a types of COCO dataset labels. Some models are trained on the COCO dataset with 80 classes (2014), and some models are trained on the COCO dataset with 91 classes (2017). Also, sometimes labels are indexed from 1, and sometimes from 0.

Imagenet

common

enum Imagenet : Enum<Imagenet>

Type of Imagenet dataset labels. Some models are trained on the vanilla ImageNet dataset, which contains 1000 classes, but some models also output additional 'background' class.

Properties

cocoCategories2014

common

val cocoCategories2014: Map<Int, String>

80 object categories in COCO dataset.

cocoCategories2017

common

val cocoCategories2017: Map<Int, String>

80 object categories in COCO dataset.

imagenetLabels

common

val imagenetLabels: Map<Int, String>

Map of Imagenet labels.