Merge pull request #37 from lorenzoh/lo/fix-inplace-project

Fix `project!` for image and mask

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.5"
+version = "0.1.6"
 
 [deps]
 ColorBlendModes = "60508b50-96e1-4007-9d6c-f475c410f16b"

src/items/image.jl

@@ -83,12 +83,11 @@ 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!(
-        buf,
+    a = OffsetArray(parent(itemdata(bufimage)), indices)
+    res = warp!(
+        a,
         box_extrapolation(itemdata(image), zero(T)),
         inv(P),
     )
-    return Image(buf, P.(getbounds(image)))
+    return Image(res, P.(getbounds(image)))
 end

src/items/mask.jl

@@ -51,13 +51,14 @@ end
 
 
 function project!(bufmask::MaskMulti, P, mask::MaskMulti, indices)
+    a = OffsetArray(parent(itemdata(bufmask)), indices)
     bounds_ = P.(getbounds(mask))
-    warp!(
-        itemdata(bufmask),
+    res = warp!(
+        a,
         mask_extrapolation(itemdata(mask)),
         inv(P),
     )
-    return MaskMulti(itemdata(bufmask), mask.classes, P.(getbounds(mask)))
+    return MaskMulti(a, mask.classes, P.(getbounds(mask)))
 end
 
 
@@ -115,12 +116,13 @@ end
 
 function project!(bufmask::MaskBinary, P, mask::MaskBinary, indices)
     bounds_ = P.(getbounds(mask))
-    warp!(
-        itemdata(bufmask),
+    a = OffsetArray(parent(itemdata(bufmask)), indices)
+    res = warp!(
+        a,
         mask_extrapolation(itemdata(mask)),
         inv(P),
     )
-    return MaskBinary(itemdata(bufmask), P.(getbounds(mask)))
+    return MaskBinary(res, P.(getbounds(mask)))
 end
 
 function showitem!(img, mask::MaskBinary)