skainet-io-core/sk.ainet.io.model/ClassNames

ClassNames

Common class name sets for object detection and classification models.

These are standard dataset class names that can be used when model metadata doesn't include class names, or for validation purposes.

Usage:

// Get COCO class names
val names = ClassNames.coco()

// Look up class name by index
val className = names.getOrNull(5) // "bus"

// Find class index by name
val classId = names.indexOf("person") // 0

Functions

coco

fun coco(): List<String>

COCO 2017 dataset class names (80 classes).

cocoMap

fun cocoMap(): Map<Int, String>

COCO 2017 class names as a map (classId -> className).

imagenet1k

fun imagenet1k(): List<String>?

ImageNet 2012 class names (1000 classes). Returns null if not available (to be implemented).

voc

fun voc(): List<String>

VOC 2012 dataset class names (20 classes).

vocMap

fun vocMap(): Map<Int, String>

VOC 2012 class names as a map.