FromFolders

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

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>)
Content copied to clipboard

Functions

getLabel
Link copied to clipboard
open override fun getLabel(dataSource: File): Float
Content copied to clipboard

Returns a label for the provided dataSource.

Properties

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

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