Imagenet

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.

Entries

V1001

common

V1001()

ImageNet labels with 1001 classes, where the 0th class is 'background'.

V1k

common

V1k()

Vanilla ImageNet labels with 1000 classes.

Functions

labels

common

fun labels(zeroIndexed: Boolean = true): Map<Int, String>

Returns map of Imagenet labels according to the Imagenet version.

Properties

name

common

val name: String

ordinal

common

val ordinal: Int