Merge pull request #61 from saddam213/Upscaler

Image Upscale Project

Author: saddam213

OnnxStack.Console/Examples/StableDiffusionGif.cs

@@ -1,4 +1,5 @@
-using OnnxStack.StableDiffusion;
+using OnnxStack.Core.Image;
+using OnnxStack.StableDiffusion;
 using OnnxStack.StableDiffusion.Common;
 using OnnxStack.StableDiffusion.Config;
 using OnnxStack.StableDiffusion.Enums;

OnnxStack.Console/Examples/UpscaleExample.cs

@@ -0,0 +1,52 @@
+using OnnxStack.Core.Image;
+using OnnxStack.ImageUpscaler.Services;
+using SixLabors.ImageSharp;
+using SixLabors.ImageSharp.PixelFormats;
+
+namespace OnnxStack.Console.Runner
+{
+    public sealed class UpscaleExample : IExampleRunner
+    {
+        private readonly string _outputDirectory;
+        private readonly IUpscaleService _imageUpscaleService;
+
+
+        public UpscaleExample(IUpscaleService imageUpscaleService)
+        {
+            _imageUpscaleService = imageUpscaleService;
+            _outputDirectory = Path.Combine(Directory.GetCurrentDirectory(), "Examples", nameof(UpscaleExample));
+            Directory.CreateDirectory(_outputDirectory);
+        }
+
+        public string Name => "Image Upscale Demo";
+
+        public string Description => "Upscales images";
+
+        public async Task RunAsync()
+        {
+            var modelSet = _imageUpscaleService.ModelSets.FirstOrDefault(x => x.Name == "RealSR BSRGAN x4");
+
+
+
+            OutputHelpers.WriteConsole("Enter Image Path", ConsoleColor.Yellow);
+            var imageFile = OutputHelpers.ReadConsole(ConsoleColor.Gray);
+            if (!File.Exists(imageFile))
+            {
+                OutputHelpers.WriteConsole("File not found!", ConsoleColor.Red);
+                return;
+            }
+
+            OutputHelpers.WriteConsole("Loading Model...", ConsoleColor.Cyan);
+            await _imageUpscaleService.LoadModelAsync(modelSet);
+            OutputHelpers.WriteConsole("Model Loaded.", ConsoleColor.Cyan);
+
+            var inputImage = await Image.LoadAsync<Rgba32>(imageFile);
+
+            OutputHelpers.WriteConsole("Upscaling Image...", ConsoleColor.Cyan);
+            var result = await _imageUpscaleService.GenerateAsImageAsync(modelSet, new InputImage(inputImage));
+            await result.SaveAsPngAsync(Path.Combine(_outputDirectory, "Result.png"));
+            OutputHelpers.WriteConsole("Upscaling Complete.", ConsoleColor.Cyan);
+        }
+
+    }
+}

OnnxStack.Console/OnnxStack.Console.csproj

@@ -15,6 +15,7 @@
   </ItemGroup>
 
   <ItemGroup>
+    <ProjectReference Include="..\OnnxStack.ImageUpscaler\OnnxStack.ImageUpscaler.csproj" />
     <ProjectReference Include="..\OnnxStack.StableDiffusion\OnnxStack.StableDiffusion.csproj" />
   </ItemGroup>
 

OnnxStack.Console/Program.cs

@@ -2,6 +2,8 @@
 using Microsoft.Extensions.Hosting;
 using Microsoft.Extensions.Logging;
 using OnnxStack.Core;
+using OnnxStack.ImageUpscaler;
+using OnnxStack.ImageUpscaler.Config;
 using System.Reflection;
 
 namespace OnnxStack.Console
@@ -17,6 +19,7 @@ static async Task Main(string[] _)
             // Add OnnxStack
             builder.Services.AddOnnxStack();
             builder.Services.AddOnnxStackStableDiffusion();
+            builder.Services.AddOnnxStackImageUpscaler();
 
             // Add AppService
             builder.Services.AddHostedService<AppService>();