ModelShape
class ModelShape @JvmOverloads constructor(var height: Int = 256, var width: Int = 144, var modelRangeMin: Int = 0, var modelRangeMax: Int = 1, var channels: Int = 4)
ModelShape Defines the shape of an ML model. This can be used to define the input and output shape of an ML model.
Parameters
height
: Int The number of pixels associated with the height of the input image, defaults to 256.
width
: Int The number of pixels associated with the width of the input image, defaults to 144.
modelRangeMin
: Int The minimum value associated with the model output, defaults to 0.
modelRangeMax
: Int The maximum value associated with the model output, defaults to 1.
channels
: Int The number of channels associated with the pixels, defaults to 4 (RGBA).