Merge pull request #497 from WilhelmusLab/496-imcomplement

feat: imcomplement
WilhelmusLab · Nov 8, 2024 · 3e74b31 · 3e74b31
2 parents fd8c4a3 + d86d8d2
commit 3e74b31
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/src/IceFloeTracker.jl b/src/IceFloeTracker.jl
@@ -71,6 +71,7 @@ include("branch.jl")
 include("special_strels.jl")
 include("tilingutils.jl")
 include("histogram_equalization.jl")
+include("imcomplement.jl")
 
 
 const sk_measure = PyNULL()

diff --git a/src/imcomplement.jl b/src/imcomplement.jl
@@ -0,0 +1,7 @@
+function imcomplement(img::Matrix{T}) where {T<:Union{Unsigned,Int}}
+    return 255 .- img
+end
+
+function imcomplement(img::Matrix{Gray{Float64}})
+    return 1 .- img
+end
diff --git a/test/test-imcomplement.jl b/test/test-imcomplement.jl
@@ -0,0 +1,9 @@
+using IceFloeTracker: imcomplement
+
+@testset "imcomplement" begin
+    img = rand(0:255, 10, 10)
+    @test imcomplement(img) == 255 .- img
+
+    img = Gray.(img ./ 255)
+    @test imcomplement(img) == 1 .- img
+end