Package org.jetbrains.kotlinx.dl.dataset.generator

Types

EmptyLabels
Link copied to clipboard
class EmptyLabels<D> : LabelGenerator<D>

This LabelGenerator is responsible for creation default labels with value Float.NaN.

FromFolders
Link copied to clipboard
class FromFolders(mapping: Map<String, Int>) : LabelGenerator<File>

This LabelGenerator is responsible for extracting labels from the names of folders where the images of the appropriate class are located. It keeps the mapping name of classes to int numbers.

LabelGenerator
Link copied to clipboard
interface LabelGenerator<D>

A parent interface for all label generators.