Add basic functions for colorspace transforms.

tests/colors/conversions.jl

@@ -1,30 +1,67 @@
 @testset "Colorspace transforms tests" begin
-    f = Chain(x->HSV.(x),channelify,flatten,Dense(768,16),Dense(16,10),x->σ.(x))
-    g_3 = Chain(Conv((3,3),7=>4,relu),
-              Conv((3,3),4=>3,relu),
-              x->colorify(RGB,x),
-              x->HSV.(x),
-              channelify,
-              flatten,
-              Dense(256*3,16),
-              Dense(16,10),
-              x->σ.(x))
-    g_1 = Chain(Conv((3,3),7=>4,relu),
-              Conv((3,3),4=>3,relu),
-              x->colorify(RGB,x),
-              x->Gray.(x),
-              channelify,
-              flatten,
-              Dense(256*1,16),
-              Dense(16,10),
-              x->σ.(x))
+    # f = Chain(x->HSV.(x),channelify,flatten,Dense(768,16),Dense(16,10),x->σ.(x))
+    # g_3 = Chain(Conv((3,3),7=>4,relu),
+    #           Conv((3,3),4=>3,relu),
+    #           x->colorify(RGB,x),
+    #           x->HSV.(x),
+    #           channelify,
+    #           flatten,
+    #           Dense(256*3,16),
+    #           Dense(16,10),
+    #           x->σ.(x))
+    # g_1 = Chain(Conv((3,3),7=>4,relu),
+    #           Conv((3,3),4=>3,relu),
+    #           x->colorify(RGB,x),
+    #           x->Gray.(x),
+    #           channelify,
+    #           flatten,
+    #           Dense(256*1,16),
+    #           Dense(16,10),
+    #           x->σ.(x))
+    ds = (7,7,5)
+    working_cspaces = (:YIQ,
+                       :LCHab,
+                       :Lab, 
+                       :BGRA,:ABGR,:BGR, 
+                       :RGBA,:ARGB,:RGB, 
+                       :HSL, 
+                       :AGray,:GrayA,:Gray, 
+                       :HSV) # colorspaces those have samplers defined for Base.Random
 
+    cspaces = (:HSV,:AHSV,:HSVA,
+               :Gray,:AGray,:GrayA,
+               :HSL,:AHSL,:HSLA,
+               :RGB,:ARGB,:RGBA,
+               :BGR,:ABGR,:BGRA,
+               :XYZ,:AXYZ,:XYZA,
+               :xyY,:AxyY,:xyYA,
+               :Lab,:ALab,:LabA,
+               :Luv,:ALuv,:LuvA,
+               :LCHab,:ALCHab,:LCHabA,
+               :LCHuv,:ALCHuv,:LCHuvA,
+               :DIN99,:ADIN99,:DIN99A,
+               :LMS,:ALMS,:LMSA,
+               :YIQ,:AYIQ,:YIQA) # tuple of all exprs of Colorspaces and their transparent variants
+    i_2 = rand(BGR, 5, 3) # 2-dim input testing
+    i_4 = rand(RGBA, 9, 9, 2) # 4-channel testing; RGBA chosen because Random sampler available
     @testset "Testing channelify" begin
-        @test size(f(rand(RGB,16,16,3))) == (10,3)
-        @test size(f(rand(Gray,16,16,3))) == (10,3)
-    end
-    @testset "Testing colorify" begin
-        @test size(g_3(rand(20,20,7,3))) == (10,3)
-        @test size(g_1(rand(20,20,7,3))) == (10,3)
+        @test size(channelify(i_2)) == (5,3,3) # to check if 2-dim inputs are working. Currently no use of them.
+        @test size(channelify(i_4)) == (9,9,4,2) # to check if 4 channel outputs are correct.
+        for cs in working_cspaces
+            i = rand(eval(cs), ds...)
+            if cs ∈ (:Gray,)
+                @test size(channelify(i)) == (ds[1:end-1]...,1,ds[end])
+            elseif cs ∈ (:AGray,:GrayA)
+                @test size(channelify(i)) == (ds[1:end-1]...,2,ds[end])
+            elseif cs ∈ (:BGRA,:ABGR,:RGBA,:ARGB)
+                @test size(channelify(i)) == (ds[1:end-1]...,4,ds[end])
+            else
+                @test size(channelify(i)) == (ds[1:end-1]...,3,ds[end])
+            end
+        end
     end
+    # @testset "Testing colorify" begin
+    #     @test size(g_3(rand(20,20,7,3))) == (10,3)
+    #     @test size(g_1(rand(20,20,7,3))) == (10,3)
+    # end
 end