Add basic functions for colorspace transforms.

.gitignore

@@ -22,3 +22,4 @@ docs/site/
 # committed for packages, but should be committed for applications that require a static
 # environment.
 Manifest.toml
+.vscode

Project.toml

@@ -4,7 +4,8 @@ authors = ["SomTambe <[email protected]>"]
 version = "0.1.0"
 
 [deps]
-CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
+ChainRules = "082447d4-558c-5d27-93f4-14fc19e9eca2"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
+ImageTransformations = "02fcd773-0e25-5acc-982a-7f6622650795"
 Images = "916415d5-f1e6-5110-898d-aaa5f9f070e0"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"

docs/make.jl

@@ -1,3 +1,33 @@
 using Documenter, DiffImages
 
-makedocs(sitename="DiffImages.jl")
+makedocs(;
+    modules = [DiffImages],
+    doctest = true,
+    authors = "Som V Tambe <[email protected]>",
+    repo = "https://github.com/SomTambe/DiffImages.jl/blob/{commit}{path}#L{line}",
+    sitename = "DiffImages.jl",
+    # format = Documenter.HTML(;
+    #     prettyurls = get(ENV, "CI", "false") == "true",
+    #     canonical = "https://fluxml.ai/Flux3D.jl",
+    #     assets = String["assets/favicon.ico"],
+    #     analytics = "UA-154580699-2",
+    # ),
+    pages = [
+        "Home" => "index.md",
+        # "Tutorials" => [
+        # ],
+        "Colorspace Transforms" => "colors/index.md",
+        # "Datasets" =>
+        #     ["ModelNet" => "datasets/modelnet.md", "Custom Dataset" => "datasets/utils.md"],
+        # "Transforms" => "api/transforms.md",
+        # "Metrics" => "api/metrics.md",
+        # "API Documentation" => [
+        #     "Conversions" => "api/conversions.md",
+        #     "Helper function" => "api/utils.md",
+        #     "Visualization" => "api/visualize.md",
+        #     "3D Models" => "api/models.md",
+        # ],
+    ],
+)
+
+deploydocs(; repo = "github.com/FluxML/Flux3D.jl.git")

docs/src/colors/index.md

@@ -0,0 +1,14 @@
+```@meta
+CurrentModule = DiffImages
+```
+# Colorspace Transforms
+
+!!! note "Consistency with the batch dimension"
+    Since we require the last dimension to be the batch dimension, 
+    kindly unsqueeze the last dimension if you would like to pass a
+    single image.
+
+```@autodocs
+Modules = [DiffImages]
+Pages = ["colors/conversions.jl"]
+```

docs/src/index.md

@@ -1,10 +1,13 @@
 ```@meta
 CurrentModule = DiffImages
 ```
-# DiffImages Documentation
 
+# DiffImages: Extending Automatic Differentiation support to the JuliaImages ecosystem
 
-```@autodocs
-Modules = [DiffImages]
-Pages = ["conversions.jl"]
-```
+DiffImages.jl is a differentiable computer vision library made in pure Julia. It is being developed with the aim of extending AD support to the JuliaImages ecosystem. 
+
+This library is under heavy development and is not intended for use yet. A stable release later will enable the users to readily use this.
+
+You can watch our development at the Discussions tab of our repository. We welcome all and everyone who intend to help us with their discussions. Open Source code simply cannot be made if the community does not give its inputs.
+
+We hope you remain safe and healthy.

src/DiffImages.jl

@@ -2,12 +2,13 @@ module DiffImages
 
 using Flux,
       Images,
-      Zygote,
-      CUDA
+      Zygote
 
 using Flux:@functor, unsqueeze
+using ImageTransformations
 
-export colorify!
-include("conversions.jl")
+export colorify, channelify
+include("colors/conversions.jl")
+include("geometry/warp.jl")
 
 end # module

src/colors/conversions.jl

@@ -0,0 +1,45 @@
+"""
+    channelify(m::AbstractArray{<:Color})
+Channelify function.
+
+Input should be in ``WHN`` order ``\\implies (*, batch)``.
+Output will be in the order ``(*, channels, batch)``.
+
+"""
+function channelify(m::AbstractArray{T}) where T <: Color
+    m = channelview(m)
+    t = ntuple(i->i, ndims(m))
+    if T <: AbstractGray
+        m = unsqueeze(m, t[end-1])
+    end
+    m = permutedims(m, (t[2:end-1]...,1,t[end]))
+    return m
+end
+
+"""
+    colorify(color::Type{<:Color}, m::AbstractArray)
+Colorify function.
+
+Expecting an input of the type ``(*, channels, batch)``.
+Converts the array to the `color` specified.
+
+
+# Examples
+```jldoctest; setup = :(using Images)
+julia> img = rand(25,25,3,7)
+
+julia> colorify(HSV, img)
+
+julia> channelify(colorify(RGB,img))==img
+true
+```
+"""
+function colorify(color::Type{<:Color}, m::AbstractArray)
+    t = ntuple(i->i, ndims(m))
+    m = permutedims(m, (t[end-1],t[1:end-2]...,t[end]))
+    if color <: AbstractGray
+        m = dropdims(m; dims=1)
+    end
+    m = colorview(color, m)
+    return m
+end

tests/runtests.jl

@@ -1,4 +1,3 @@
 using Test,
       Zygote,
-      Flux,
-      CUDA
+      Flux