search

FluxML / DataAugmentation.jl

Flexible data augmentation library for machine and deep learning
https://fluxml.ai/DataAugmentation.jl/dev/
MIT License
41 stars 17 forks source link

ImageToTensor doesn't support RGBA #83

Open oluwandabira opened 11 months ago

oluwandabira commented 11 months ago

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