FromFolders

class FromFolders(mapping: Map<String, Int>) : LabelGenerator

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.

Constructors

FromFolders
Link copied to clipboard
fun FromFolders(mapping: Map<String, Int>)

Properties

mapping
Link copied to clipboard
val mapping: Map<String, Int>

The mapping from class names to class labels presented as natural numbers.