Update ImageUpscale to new API format

OnnxStack.Console/Examples/UpscaleExample.cs

@@ -1,18 +1,14 @@
 using OnnxStack.Core.Image;
 using OnnxStack.FeatureExtractor.Pipelines;
-using OnnxStack.ImageUpscaler.Config;
-using SixLabors.ImageSharp;
 
 namespace OnnxStack.Console.Runner
 {
  public sealed class UpscaleExample : IExampleRunner
  {
  private readonly string _outputDirectory;
- private readonly ImageUpscalerConfig _configuration;
 
- public UpscaleExample(ImageUpscalerConfig configuration)
+ public UpscaleExample()
  {
- _configuration = configuration;
  _outputDirectory = Path.Combine(Directory.GetCurrentDirectory(), "Examples", nameof(UpscaleExample));
  Directory.CreateDirectory(_outputDirectory);
  }

OnnxStack.Console/Program.cs

@@ -19,7 +19,6 @@ static async Task Main(string[] _)
  // Add OnnxStack
  builder.Services.AddOnnxStack();
  builder.Services.AddOnnxStackConfig<StableDiffusionConfig>();
- builder.Services.AddOnnxStackImageUpscaler();
 
  // Add AppService
  builder.Services.AddHostedService<AppService>();

OnnxStack.ImageUpscaler/Extensions/ImageExtensions.cs

@@ -18,7 +18,7 @@ internal static class ImageExtensions
  /// <param name="sampleSize">Maximum size of the tile.</param>
  /// <param name="scaleFactor">The scale factor.</param>
  /// <returns></returns>
- public static List<ImageTile> GenerateTiles(this OnnxImage imageSource, int sampleSize, int scaleFactor)
+ internal static List<ImageTile> GenerateTiles(this OnnxImage imageSource, int sampleSize, int scaleFactor)
  {
  var tiles = new List<ImageTile>();
  var tileSizeX = Math.Min(sampleSize, imageSource.Width);
@@ -48,7 +48,7 @@ public static List<ImageTile> GenerateTiles(this OnnxImage imageSource, int samp
  /// <param name="sourceImage">The source image.</param>
  /// <param name="sourceArea">The source area.</param>
  /// <returns></returns>
- public static OnnxImage ExtractTile(this OnnxImage sourceImage, Rectangle sourceArea)
+ internal static OnnxImage ExtractTile(this OnnxImage sourceImage, Rectangle sourceArea)
  {
  var height = sourceArea.Height;
  var targetImage = new Image<Rgba32>(sourceArea.Width, sourceArea.Height);
@@ -65,7 +65,7 @@ public static OnnxImage ExtractTile(this OnnxImage sourceImage, Rectangle source
  }
 
 
- public static void ApplyImageTile(this DenseTensor<float> imageTensor, DenseTensor<float> tileTensor, Rectangle location)
+ internal static void ApplyImageTile(this DenseTensor<float> imageTensor, DenseTensor<float> tileTensor, Rectangle location)
  {
  var offsetY = location.Y;
  var offsetX = location.X;

OnnxStack.ImageUpscaler/Models/ImageTile.cs

@@ -3,7 +3,7 @@
 
 namespace OnnxStack.ImageUpscaler.Models
 {
- public record ImageTile
+ internal record ImageTile
  {
  public OnnxImage Image { get; set; }
  public Rectangle Destination { get; set; }

OnnxStack.ImageUpscaler/Models/UpscaleInput.cs

@@ -2,7 +2,7 @@
 
 namespace OnnxStack.ImageUpscaler.Models
 {
- public record UpscaleInput(List<ImageTile> ImageTiles, int OutputWidth, int OutputHeight);
+ internal record UpscaleInput(List<ImageTile> ImageTiles, int OutputWidth, int OutputHeight);
 }
 
 

OnnxStack.ImageUpscaler/README.md

@@ -1 +1,29 @@
-# OnnxStack.ImageUpsacler
+# OnnxStack.ImageUpscaler
+
+## Upscale Models
+https://huggingface.co/wuminghao/swinir
+https://huggingface.co/rocca/swin-ir-onnx
+https://huggingface.co/Xenova/swin2SR-classical-sr-x2-64
+https://huggingface.co/Xenova/swin2SR-classical-sr-x4-64
+
+
+# Basic Example
+```csharp
+// Load Input Image
+var inputImage = await OnnxImage.FromFileAsync("Input.png");
+
+// Create Pipeline
+var pipeline = ImageUpscalePipeline.CreatePipeline("003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x4_GAN.onnx", scaleFactor: 4);
+
+// Run pipeline
+var result = await pipeline.RunAsync(inputImage);
+
+// Create Image from Tensor result
+var image = new OnnxImage(result, ImageNormalizeType.ZeroToOne);
+
+// Save Image File
+await image.SaveAsync("Upscaled.png");
+
+// Unload
+await pipeline.UnloadAsync();
+```