Handle masks of IndirectArrays

Author: lorenzoh

Project.toml

@@ -10,6 +10,7 @@ Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 ImageDraw = "4381153b-2b60-58ae-a1ba-fd683676385f"
 ImageTransformations = "02fcd773-0e25-5acc-982a-7f6622650795"
 Images = "916415d5-f1e6-5110-898d-aaa5f9f070e0"
+IndirectArrays = "9b13fd28-a010-5f03-acff-a1bbcff69959"
 Interpolations = "a98d9a8b-a2ab-59e6-89dd-64a1c18fca59"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MosaicViews = "e94cdb99-869f-56ef-bcf0-1ae2bcbe0389"

src/DataAugmentation.jl

@@ -3,6 +3,7 @@ module DataAugmentation
 using ColorBlendModes
 using CoordinateTransformations
 using Distributions: Sampleable, Uniform, Categorical
+using IndirectArrays: IndirectArray
 using ImageDraw
 using Images
 using Images: Colorant, permuteddimsview

src/items/mask.jl

@@ -31,6 +31,8 @@ end
 
 MaskMulti(a::AbstractArray{<:Gray{T}}, args...) where T = MaskMulti(reinterpret(T, a), args...)
 MaskMulti(a::AbstractArray{<:Normed{T}}, args...) where T = MaskMulti(reinterpret(T, a), args...)
+MaskMulti(a::IndirectArray, classes = a.values, bounds = Bounds(size(a))) =
+    MaskMulti(a.index, classes, bounds)
 
 Base.show(io::IO, mask::MaskMulti{N, T}) where {N, T} =
     print(io, "MaskMulti{$N, $T}() with size $(size(itemdata(mask))) and $(length(mask.classes)) classes")