feat: parallelize gradient magnitude

WilhelmusLab · Nov 5, 2024 · 8755d9c · 8755d9c
1 parent f67913d
commit 8755d9c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/segmentation_a_direct.jl b/src/segmentation_a_direct.jl
@@ -138,7 +138,9 @@ Compute the gradient magnitude of an image using the Sobel operator.
 """
 function imgradientmag(img)
     h = centered([-1 0 1; -2 0 2; -1 0 1]')
-    Gx = imfilter(img, h', "replicate")
-    Gy = imfilter(img, h, "replicate")
+    Gx_future = Threads.@spawn IceFloeTracker.imfilter(img, h', "replicate")
+    Gy_future = Threads.@spawn IceFloeTracker.imfilter(img, h, "replicate")
+    Gx = fetch(Gx_future)
+    Gy = fetch(Gy_future)
     return hypot.(Gx, Gy)
-end
+end