Package org.jetbrains.kotlinx.dl.dataset.handler

Functions

extractCifar10Images

fun extractCifar10Images(archiveName: String): Array<FloatArray>

Loads images from archiveName to heap memory and applies basic normalization preprocessing.

extractCifar10Labels

fun extractCifar10Labels(pathToLabels: String, numClasses: Int): FloatArray

Loads labels from pathToLabels csv file to heap memory and converts to Floats.

extractCifar10LabelsAnsSort

fun extractCifar10LabelsAnsSort(pathToLabels: String, numClasses: Int): FloatArray

Loads labels from pathToLabels csv file to heap memory and converts to Floats, after that it sorts it to have the same order as image files.

extractFashionImages

fun extractFashionImages(archivePath: String): Array<FloatArray>

Extracts Fashion Mnist images from archivePath.

extractFashionLabels

fun extractFashionLabels(archivePath: String, numClasses: Int): Array<FloatArray>

Extracts Fashion Mnist labels from archivePath with number of classes numClasses.

extractImages

fun extractImages(archivePath: String): Array<FloatArray>

Extracts (Fashion) Mnist images from archivePath.

extractLabels

fun extractLabels(archivePath: String, numClasses: Int): FloatArray

Extracts (Fashion) Mnist labels from archivePath with number of classes numClasses.

Properties

cocoCategories

val cocoCategories: Map<Int, String>

80 object categories in COCO dataset.

cocoCategoriesForSSD

val cocoCategoriesForSSD: Map<Int, String>

80 object categories in COCO dataset.

FASHION_TEST_IMAGES_ARCHIVE

const val FASHION_TEST_IMAGES_ARCHIVE: String

Path to test images archive of Fashion Mnist Dataset.

FASHION_TEST_LABELS_ARCHIVE

const val FASHION_TEST_LABELS_ARCHIVE: String

Path to test labels archive of Fashion Mnist Dataset.

FASHION_TRAIN_IMAGES_ARCHIVE

const val FASHION_TRAIN_IMAGES_ARCHIVE: String

Path to train images archive of Fashion Mnist Dataset.

FASHION_TRAIN_LABELS_ARCHIVE

const val FASHION_TRAIN_LABELS_ARCHIVE: String

Path to train labels archive of Fashion Mnist Dataset.

NUMBER_OF_CLASSES

const val NUMBER_OF_CLASSES: Int = 10

Number of classes.

TEST_IMAGES_ARCHIVE

const val TEST_IMAGES_ARCHIVE: String

Path to test images archive of Mnist Dataset.

TEST_LABELS_ARCHIVE

const val TEST_LABELS_ARCHIVE: String

Path to test labels archive of Mnist Dataset.

TRAIN_IMAGES_ARCHIVE

const val TRAIN_IMAGES_ARCHIVE: String

Path to train images archive of Mnist Dataset.

TRAIN_LABELS_ARCHIVE

const val TRAIN_LABELS_ARCHIVE: String

Path to train labels archive of Mnist Dataset.