ImageTensor GetPixel/SetPixel support

Author: saddam213

TensorStack.Common/Extensions/Extensions.cs

@@ -167,5 +167,27 @@ public static async Task SafeCancelAsync(this CancellationTokenSource cancellati
             }
             catch { }
         }
+
+
+        /// <summary>
+        /// Normalizes to float.
+        /// </summary>
+        /// <param name="value">The value.</param>
+        /// <returns>System.Single.</returns>
+        public static float NormalizeToFloat(this byte value)
+        {
+            return (value / 255.0f) * 2.0f - 1.0f;
+        }
+
+
+        /// <summary>
+        /// Denormalizes to byte.
+        /// </summary>
+        /// <param name="value">The value.</param>
+        /// <returns>System.Byte.</returns>
+        public static byte DenormalizeToByte(this float value)
+        {
+            return (byte)Math.Round(Math.Clamp((value / 2.0 + 0.5) * 255.0, 0.0, 255.0));
+        }
     }
 }

TensorStack.Common/Image/ImagePixel.cs

@@ -0,0 +1,18 @@
+namespace TensorStack.Common.Image
+{
+    public readonly struct ImagePixel
+    {
+        public readonly float R;
+        public readonly float G;
+        public readonly float B;
+        public readonly float A;
+
+        public ImagePixel(float r, float g, float b, float a)
+        {
+            R = r;
+            G = g;
+            B = b;
+            A = a;
+        }
+    }
+}

TensorStack.Common/Tensor/ImageTensor.cs

@@ -1,6 +1,7 @@
 // Copyright (c) TensorStack. All rights reserved.
 // Licensed under the Apache 2.0 License.
 using System;
+using TensorStack.Common.Image;
 
 namespace TensorStack.Common.Tensor
 {
@@ -16,7 +17,7 @@ public class ImageTensor : Tensor<float>
         /// </summary>
         /// <param name="tensor">The tensor.</param>
         public ImageTensor(Tensor<float> tensor)
-            : base(ProcessChannels(tensor), [1, 4, ..tensor.Dimensions[^2..]])
+            : base(ProcessChannels(tensor), [1, 4, .. tensor.Dimensions[^2..]])
         {
             ThrowIfInvalid();
         }
@@ -175,6 +176,55 @@ public ImageTensor CloneAs()
             return Clone().AsImageTensor();
         }
 
+
+        /// <summary>
+        /// Gets the RGBA pixel from ImageTensor
+        /// </summary>
+        /// <param name="imageTensor">The image tensor.</param>
+        /// <param name="x">The x.</param>
+        /// <param name="y">The y.</param>
+        /// <returns>ImagePixel.</returns>
+        /// <exception cref="System.ArgumentOutOfRangeException">Pixel ({x},{y}) out of range ({imageTensor.Width}x{imageTensor.Height}).</exception>
+        public ImagePixel GetPixel(int x, int y)
+        {
+            if (x >= Width || y >= Height)
+                throw new ArgumentOutOfRangeException($"Pixel ({x},{y}) out of range ({Width}x{Height}).");
+
+            var pixelIndex = y * Width + x;
+            var span = Memory.Span;
+            var stride = Height * Width;
+            return new ImagePixel(
+                span[0 * stride + pixelIndex],
+                span[1 * stride + pixelIndex],
+                span[2 * stride + pixelIndex],
+                span[3 * stride + pixelIndex]
+            );
+        }
+
+
+        /// <summary>
+        /// Sets the RGBA pixel for ImageTensor
+        /// </summary>
+        /// <param name="imageTensor">The image tensor.</param>
+        /// <param name="x">The x.</param>
+        /// <param name="y">The y.</param>
+        /// <param name="color">The color.</param>
+        /// <exception cref="System.ArgumentOutOfRangeException">Pixel ({x},{y}) out of range ({imageTensor.Width}x{imageTensor.Height}).</exception>
+        public void SetPixel(int x, int y, ImagePixel color)
+        {
+            if (x >= Width || y >= Height)
+                throw new ArgumentOutOfRangeException($"Pixel ({x},{y}) out of range ({Width}x{Height}).");
+
+            int pixelIndex = y * Width + x;
+            var span = Memory.Span;
+            int stride = Height * Width;
+            span[0 * stride + pixelIndex] = color.R;
+            span[1 * stride + pixelIndex] = color.G;
+            span[2 * stride + pixelIndex] = color.B;
+            span[3 * stride + pixelIndex] = color.A;
+        }
+
+
         /// <summary>
         /// Throws if Dimensions are invalid.
         /// </summary>

TensorStack.Image.Bitmap/Extensions.cs

@@ -3,6 +3,7 @@
 using System.Drawing;
 using System.Drawing.Imaging;
 using System.Threading.Tasks;
+using TensorStack.Common;
 using TensorStack.Common.Tensor;
 
 namespace TensorStack.Image

TensorStack.Image.BitmapImage/Extensions.cs

@@ -5,6 +5,7 @@
 using System.Windows;
 using System.Windows.Media;
 using System.Windows.Media.Imaging;
+using TensorStack.Common;
 using TensorStack.Common.Tensor;
 
 namespace TensorStack.Image

TensorStack.Image.ImageSharp/Extensions.cs

@@ -3,6 +3,7 @@
 using SixLabors.ImageSharp;
 using SixLabors.ImageSharp.PixelFormats;
 using System.Threading.Tasks;
+using TensorStack.Common;
 using TensorStack.Common.Tensor;
 
 namespace TensorStack.Image