ImageToTensor doesn't support RGBA

[oluwandabira]

Minimal example:

using DataAugmentation, Images

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

which results in

ERROR: LoadError: MethodError: no method matching imagetotensor(::Matrix{RGBA{Float64}}, ::Type{Float32})

Closest candidates are:
  imagetotensor(::AbstractArray{C, N}, ::Any) where {C<:Color, N}

Pkg.status():

[88a5189c] DataAugmentation v0.2.11
[916415d5] Images v0.26.0

[paulnovo]

@oluwandabira I think this has already been fixed by relaxing the type from Color to Colorant in imagetotensor.

With DataAugmentation v0.3.2:

julia> using DataAugmentation, Colors, Images

julia> RGBA<:Color
false

julia> RGBA<:Colorant
true

julia> image = Image(rand(RGBA, 50, 50))
Image{2, RGBA{Float64}}() with bounds Bounds(1:50×1:50)

julia> tfm = ImageToTensor()
ImageToTensor{Float32}()

julia> apply(tfm, image)
ArrayItem{3, Float32}() of size (50, 50, 4)