OnFlyImageDataset

class OnFlyImageDataset : Dataset
Content copied to clipboard

This dataset keeps all data on disk and generates batches on-fly according preprocessing pipeline.

Parameters

xFiles

files to load images from

y

labels to use for the loaded images

preprocessing

preprocessing to apply to the loaded images

Types

Companion
Link copied to clipboard
object Companion
Content copied to clipboard

Functions

batchIterator
Link copied to clipboard
fun batchIterator(batchSize: Int): Dataset.BatchIterator
Content copied to clipboard

Returns BatchIterator with fixed batchSize.

getX
Link copied to clipboard
open override fun getX(idx: Int): FloatArray
Content copied to clipboard

Returns row by index idx.

getY
Link copied to clipboard
open override fun getY(idx: Int): Float
Content copied to clipboard

Returns label as FloatArray by index idx.

shuffle
Link copied to clipboard
open override fun shuffle(): OnFlyImageDataset
Content copied to clipboard

Shuffles the dataset.

split
Link copied to clipboard
open override fun split(splitRatio: Double): Pair<OnFlyImageDataset, OnFlyImageDataset>
Content copied to clipboard

Splits datasets on two sub-datasets according splitRatio.

xSize
Link copied to clipboard
open override fun xSize(): Int
Content copied to clipboard

Returns amount of data rows.