Merge branch 'master' of https://github.com/lorenzoh/DataAugmentation.jl

Author: lorenzoh

Project.toml

@@ -1,7 +1,7 @@
 name = "DataAugmentation"
 uuid = "88a5189c-e7ff-4f85-ac6b-e6158070f02e"
 authors = ["lorenzoh <lorenz.ohly@gmail.com>"]
-version = "0.1.1"
+version = "0.1.3"
 
 [deps]
 ColorBlendModes = "60508b50-96e1-4007-9d6c-f475c410f16b"
@@ -11,6 +11,7 @@ ImageTransformations = "02fcd773-0e25-5acc-982a-7f6622650795"
 Images = "916415d5-f1e6-5110-898d-aaa5f9f070e0"
 Interpolations = "a98d9a8b-a2ab-59e6-89dd-64a1c18fca59"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
 Parameters = "d96e819e-fc66-5662-9728-84c9c7592b0a"
 Rotations = "6038ab10-8711-5258-84ad-4b1120ba62dc"
 Setfield = "efcf1570-3423-57d1-acb7-fd33fddbac46"

src/DataAugmentation.jl

@@ -8,6 +8,7 @@ using Images: Colorant, permuteddimsview
 using ImageTransformations
 using ImageTransformations: center, _center, box_extrapolation, warp!
 using Interpolations
+using OffsetArrays: OffsetArray
 using LinearAlgebra: I
 using Parameters
 using Rotations

src/items/image.jl

@@ -81,10 +81,12 @@ end
 # as they are implicitly given by the buffer.
 
 function project!(bufimage::Image, P, image::Image{N, T}, indices) where {N, T}
+    # Update index offsets so the correct portion of the image is warped
+    buf = OffsetArray(itemdata(bufimage), indices)
     warp!(
-        itemdata(bufimage),
+        buf,
         box_extrapolation(itemdata(image), zero(T)),
         inv(P),
     )
-    return Image(itemdata(bufimage), P.(getbounds(image)))
+    return Image(buf, P.(getbounds(image)))
 end