ImageShape

data class ImageShape(width: Long?, height: Long?, channels: Long?)

Helper class to keep widely used shape of image object presented as a 4D tensor (batchSize = 1, width, height, channels). null values are allowed for width, height and channels, indicating that the dimension size is unknown.

Constructors

ImageShape

fun ImageShape(width: Long? = null, height: Long? = null, channels: Long? = null)

Properties

channels

val channels: Long? = null

number of channels in the image

height

val height: Long? = null

image height

numberOfElements

val numberOfElements: Long

Returns number of elements in a tensor with the given shape.

width

val width: Long? = null

image width