ImageToTensor()

Expands an Image{N, T} of size sz to an ArrayItem{N+1} with size (sz..., ch) where ch is the number of color channels of T.

Supports apply!.

Examples

using DataAugmentation, Images

image = Image(rand(RGB, 50, 50))
tfm = ImageToTensor()
apply(tfm, image)

ArrayItem{3, Float32}() of size (50, 50, 3)