Package org.jetbrains.kotlinx.dl.dataset.embedded

Functions

cifar10Paths
Link copied to clipboard
fun cifar10Paths(cacheDirectory: File = File("cache")): Pair<String, String>

Returns paths to images and its labels for the Cifar'10 dataset.

dogsCatsDatasetPath
Link copied to clipboard
fun dogsCatsDatasetPath(cacheDirectory: File = File("cache")): String

Returns path to images of the Dogs-vs-Cats dataset.

dogsCatsSmallDatasetPath
Link copied to clipboard
fun dogsCatsSmallDatasetPath(cacheDirectory: File = File("cache")): String

Returns path to images of the subset of the Dogs-vs-Cats dataset.

extractCifar10Images
Link copied to clipboard
fun extractCifar10Images(archiveName: String): Array<FloatArray>

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

extractCifar10Labels
Link copied to clipboard
fun extractCifar10Labels(pathToLabels: String, numClasses: Int): FloatArray

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

extractCifar10LabelsAnsSort
Link copied to clipboard
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
Link copied to clipboard
fun extractFashionImages(archivePath: String): Array<FloatArray>

Extracts Fashion Mnist images from archivePath.

extractFashionLabels
Link copied to clipboard
fun extractFashionLabels(archivePath: String, numClasses: Int): Array<FloatArray>

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

extractImages
Link copied to clipboard
fun extractImages(archivePath: String): Array<FloatArray>

Extracts (Fashion) Mnist images from archivePath.

extractLabels
Link copied to clipboard
fun extractLabels(archivePath: String, numClasses: Int): FloatArray

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

fashionMnist
Link copied to clipboard
fun fashionMnist(cacheDirectory: File = File("cache")): Pair<OnHeapDataset, OnHeapDataset>

Loads the Fashion-MNIST dataset.

freeSpokenDigitDatasetPath
Link copied to clipboard
fun freeSpokenDigitDatasetPath(cacheDirectory: File = File("cache")): String

Returns path to sound data files from Free Spoken Digits Dataset.

freeSpokenDigits
Link copied to clipboard
fun freeSpokenDigits(cacheDirectory: File = File("cache"), maxTestIndex: Int = 5): Pair<OnHeapDataset, OnHeapDataset>

Loads the Free Spoken Digits Dataset. This is a dataset of wav sound files of the 10 digits spoken by different people many times each. The test set officially consists of the first 10% of the recordings. Recordings numbered 0-4 (inclusive) are in the test and 5-49 are in the training set.

mnist
Link copied to clipboard
fun mnist(cacheDirectory: File = File("cache")): Pair<OnHeapDataset, OnHeapDataset>

Loads the MNIST dataset. This is a dataset of 60,000 28x28 grayscale images of the 10 digits, along with a test set of 10,000 images. More info can be found at the MNIST homepage.

mnist3D
Link copied to clipboard
fun mnist3D(cacheDirectory: File = File("cache")): Pair<OnHeapDataset, OnHeapDataset>

Loads the MNIST 3D dataset. This is a dataset of 10,000 16x16x16 grayscale 3D images of the 10 digits, along with a test set of 2,000 3D images.

Properties

FASHION_TEST_IMAGES_ARCHIVE
Link copied to clipboard
const val FASHION_TEST_IMAGES_ARCHIVE: String

Path to test images archive of Fashion Mnist Dataset.

FASHION_TEST_LABELS_ARCHIVE
Link copied to clipboard
const val FASHION_TEST_LABELS_ARCHIVE: String

Path to test labels archive of Fashion Mnist Dataset.

FASHION_TRAIN_IMAGES_ARCHIVE
Link copied to clipboard
const val FASHION_TRAIN_IMAGES_ARCHIVE: String

Path to train images archive of Fashion Mnist Dataset.

FASHION_TRAIN_LABELS_ARCHIVE
Link copied to clipboard
const val FASHION_TRAIN_LABELS_ARCHIVE: String

Path to train labels archive of Fashion Mnist Dataset.

FSDD_SOUND_DATA_SIZE
Link copied to clipboard
const val FSDD_SOUND_DATA_SIZE: Long = 20480
MNIST_3D_DATASET
Link copied to clipboard
const val MNIST_3D_DATASET: String

Path to H5 file of Mnist 3D Dataset.

NUMBER_OF_CLASSES
Link copied to clipboard
const val NUMBER_OF_CLASSES: Int = 10

Number of classes.

TEST_IMAGES_ARCHIVE
Link copied to clipboard
const val TEST_IMAGES_ARCHIVE: String

Path to test images archive of Mnist Dataset.

TEST_LABELS_ARCHIVE
Link copied to clipboard
const val TEST_LABELS_ARCHIVE: String

Path to test labels archive of Mnist Dataset.

TRAIN_IMAGES_ARCHIVE
Link copied to clipboard
const val TRAIN_IMAGES_ARCHIVE: String

Path to train images archive of Mnist Dataset.

TRAIN_LABELS_ARCHIVE
Link copied to clipboard
const val TRAIN_LABELS_ARCHIVE: String

Path to train labels archive of Mnist Dataset.