![Warlock-Studio banner](Assets/banner.png)
### _AI Media Enhancement Suite_ [![Build Status](https://img.shields.io/badge/Build-Stable_Release-blue?style=for-the-badge&logo=github)](https://github.com/Ivan-Ayub97/Warlock-Studio/releases) [![Version](https://img.shields.io/badge/Version-4.1--08.01-darkred?style=for-the-badge&logo=git)](https://github.com/Ivan-Ayub97/Warlock-Studio/releases/tag/4.1) [![License](https://img.shields.io/badge/License-MIT-purple?style=for-the-badge&logo=open-source-initiative)](LICENSE) [![Downloads](https://img.shields.io/github/downloads/Ivan-Ayub97/Warlock-Studio/total?style=for-the-badge&color=gold&logo=download)](https://github.com/Ivan-Ayub97/Warlock-Studio/releases) [![Platform](https://img.shields.io/badge/Platform-Windows?style=for-the-badge&logo=windows&logoColor=white)](#) [![Python](https://img.shields.io/badge/Python-3.10+-3776AB?style=for-the-badge&logo=python&logoColor=yellow)](https://www.python.org/) [![Issues](https://img.shields.io/github/issues/Ivan-Ayub97/Warlock-Studio?style=for-the-badge&logo=github)](https://github.com/Ivan-Ayub97/Warlock-Studio/issues) [![Last Commit](https://img.shields.io/github/last-commit/Ivan-Ayub97/Warlock-Studio?style=for-the-badge&logo=git)](https://github.com/Ivan-Ayub97/Warlock-Studio/commits/main) _Transform your media with cutting-edge AI technology_ --- **Warlock-Studio** is an open source desktop application for **Windows**, designed to integrate state-of-the-art AI models for **image and video enhancement**. Inspired by [Djdefrag](https://github.com/Djdefrag) tools such as **QualityScaler** and **FluidFrames**, Warlock-Studio provides a unified, high-performance platform for **upscaling, restoration, denoising, and frame interpolation**. Version **4.1** introduces **enhanced GPU utilization**, **robust ONNX model loading**, and multiple **stability and compatibility improvements**, ensuring a smoother, faster, and more reliable user experience. --- ## πŸ“₯ Download Installer (v4.1) Get the latest stable release from:
Download Warlock-Studio Download from GitHub Download from itch.io
--- ## ✨ Key Features - **AI Upscaling & Restoration** – Utilize **Real-ESRGAN, BSRGAN, and IRCNN** models for denoising, super-resolution, and detail recovery. - **Face Restoration (GFPGAN)** – Recover facial details from low-resolution or blurry images and video frames. - **Frame Interpolation (RIFE)** – Smooth motion or generate slow-motion content with **2Γ—, 4Γ—, or 8Γ— interpolation**. - **Modern UI** – Redesigned interface in v4.0 for **intuitive workflow and streamlined navigation**. - **Batch Processing** – Process multiple media files simultaneously, saving time and effort. - **Custom Workflows** – Fine-grained control over models, resolution, output formats, and quality parameters. - **Open-Source & Extensible** – Fully MIT licensed, for contributors and developers. --- ## πŸ†• What’s New in v4.1 - πŸ”§ **Removed outdated SuperResolution-10 model** to streamline resources. - βœ… **Enhanced ONNX loading** with better GPU acceleration. - βœ… **Fixed import and type annotation issues** for cleaner code execution. - βœ… **Improved error handling** with graceful fallback mechanisms. - 🟒 Optimized **GPU resource management** for faster processing. - πŸš€ Compatibility fixes for **NumPy** and **OpenCV**. - πŸ“¦ Stability and memory usage improvements for longer sessions. - βœ… Increased startup reliability with enhanced user notifications. --- ## 🌐 Smart Model Distribution System (v4.0+) ### 🎯 Lightweight Installation - Installer reduced from **1.4GB β†’ 450MB** (–68%) for faster downloads. - Core AI models (~400MB) downloaded automatically on first launch. - Bandwidth-friendly and modular setup for selective installation. ### πŸ›‘οΈ Reliability - **Integrity checks** on downloaded model files. - **Graceful degradation** in case of missing models. - **Offline support** allows manual placement of model files. --- ## πŸ–ΌοΈ Interface Previews **Main Window** ![Main interface](rsc/Capture.png) **RIFE Options** ![RIFE Options](rsc/CaptureRIFE.png) --- ## πŸš€ How to Use 1. Run Warlock-Studio as **Administrator** (recommended for full GPU access). 2. **Load Media** – Import images or videos. 3. **Configure Processing Settings**: - Select AI model (Real-ESRGAN, GFPGAN, etc.) - Set resolution, format, frame interpolation, and quality. 4. **Start Processing** using **"Make Magic"**. 5. Retrieve the processed results from the designated output folder. --- ## πŸ–ΌοΈ Quality Comparison Enhanced image using **BSRGANx2**: ![Comparison](rsc/image_comparison.png) --- ## πŸ“Š Model Comparison | Model File | Use Case | Speed | Quality | Notes | |--------------------------|----------------------------------------|---------|---------|-------| | **GFPGANv1.4** | Face restoration | High | High | Optimal for portraits | | **BSRGANx2** | 2Γ— upscale + denoising | Medium | Very High | Suitable for lightly degraded media | | **BSRGANx4** | 4Γ— upscale + denoising | Low | Very High | For heavily degraded content | | **RIFE** | Frame interpolation | High | High | Smooth motion, slow-motion support | | **RIFE-Lite** | Lightweight interpolation | Very High | Medium | Faster, lower resource usage | | **RealESRGANx4** | General 4Γ— upscaling | Medium | High | Balanced performance | | **RealESRNetx4** | Subtle restoration | Medium | High | Preserves natural image texture | | **RealSRx4_Anime** | Anime / line-art enhancement | Medium | High | Sharp edges for 2D art | | **IRCNN_L** | Light denoising | High | Medium | Mild artifact removal | | **IRCNN_M** | Medium denoising | High | Medium | Stronger artifact cleanup | --- ## βš™οΈ Installation 1. **Download installer** (see links above). 2. Run the **setup wizard** and follow prompts. 3. Launch via Start Menu or Desktop shortcut. > Warlock-Studio is packaged using **PyInstaller** and deployed with **Inno Setup** for seamless installation. ### Installer Previews ![Installer 1](rsc/Installation_window.png) ![Installer 2](rsc/Installation_window2.png) ![Installer 3](rsc/Installation_window3.png) --- ## πŸ–₯️ System Requirements - **OS:** Windows 10 or higher (64-bit) - **RAM:** 8GB+ recommended - **GPU:** NVIDIA or DirectML-compatible GPU recommended - **Storage:** Sufficient free space for input and processed media --- ## πŸ“Œ Development Status (v4.1-08.01) | Component | Status | Notes | |-----------------------------|------------|-------| | Upscaling Models | 🟒 Stable | Includes VRAM recovery integration | | Optimized Model Suite | 🟒 Enhanced | Streamlined and reliable | | Face Restoration (GFPGAN) | 🟒 Stable | High-quality face reconstruction | | Frame Interpolation (RIFE) | 🟒 Stable | Smooth motion and slow-motion support | | Batch Processing | 🟒 Stable | Improved error handling and logging | | User Interface (UI/UX) | 🟒 Refined | Clean, modern design with integrated models | | GPU Management | 🟒 Enhanced | Robust ONNX handling and fallbacks | | Code Quality | 🟒 Improved | Refactored, type-safe, maintainable | | Installer & Packaging | 🟒 Stable | Smooth setup experience | --- ## πŸ“‚ Project Structure
``` Warlock-Studio/ β”œβ”€β”€ AI-onnx/ # Pre-trained ONNX models for AI processing β”‚ β”œβ”€β”€ BSRGANx2_fp16.onnx β”‚ β”œβ”€β”€ BSRGANx4_fp16.onnx β”‚ β”œβ”€β”€ GFPGANv1.4.fp16.onnx β”‚ β”œβ”€β”€ IRCNN_Lx1_fp16.onnx β”‚ β”œβ”€β”€ IRCNN_Mx1_fp16.onnx β”‚ β”œβ”€β”€ RealESR_Animex4_fp16.onnx β”‚ β”œβ”€β”€ RealESR_Gx4_fp16.onnx β”‚ β”œβ”€β”€ RealESRGANx4_fp16.onnx β”‚ β”œβ”€β”€ RealESRNetx4_fp16.onnx β”‚ β”œβ”€β”€ RealSRx4_Anime_fp16.onnx β”‚ β”œβ”€β”€ RIFE_fp32.onnx β”‚ └── RIFE_Lite_fp32.onnx β”‚ β”œβ”€β”€ Assets/ # Application assets and third-party binaries β”‚ β”œβ”€β”€ banner.png β”‚ β”œβ”€β”€ clear_icon.png β”‚ β”œβ”€β”€ exiftool.exe β”‚ β”œβ”€β”€ ffmpeg.exe β”‚ β”œβ”€β”€ info_icon.png β”‚ β”œβ”€β”€ logo.ico β”‚ β”œβ”€β”€ logo.png β”‚ β”œβ”€β”€ stop_icon.png β”‚ β”œβ”€β”€ upscale_icon.png β”‚ β”œβ”€β”€ wizard-image.bmp β”‚ └── wizard-small.bmp β”‚ β”œβ”€β”€ rsc/ # UI previews, workflows, and branding resources β”‚ β”œβ”€β”€ itch.io.png β”‚ β”œβ”€β”€ Capture.png β”‚ β”œβ”€β”€ CaptureRIFE.png β”‚ β”œβ”€β”€ GitHub_Logo_WS.png β”‚ β”œβ”€β”€ WorkflowBSRGAN.png β”‚ β”œβ”€β”€ WorkflowIRCNN.png β”‚ β”œβ”€β”€ WorkflowRealESRGAN.png β”‚ β”œβ”€β”€ WorkflowRIFE.png β”‚ β”œβ”€β”€ Installation_window.png β”‚ β”œβ”€β”€ Installation_window2.png β”‚ └── Installation_window3.png β”‚ β”œβ”€β”€ Manual/ # LaTeX sources and generated manuals β”‚ β”œβ”€β”€ Manual_EN.tex β”‚ β”œβ”€β”€ Manual_ES.tex β”‚ β”œβ”€β”€ v4.1_User_Manual_EN.pdf β”‚ └── v4.1_User_Manual_ES.pdf β”‚ β”œβ”€β”€ Warlock-Studio.py # Main application script β”œβ”€β”€ Warlock-Studio.spec # PyInstaller specification file β”œβ”€β”€ Setup.iss # Inno Setup installer script β”œβ”€β”€ README.md # Project overview β”œβ”€β”€ requirements.txt # Python dependencies β”œβ”€β”€ CHANGELOG.md # Version history and updates β”œβ”€β”€ LICENSE # MIT License information β”œβ”€β”€ model_downloader.py # Still in the process of implementation β”œβ”€β”€ NOTICE.md # Legal notices and attributions β”œβ”€β”€ CODE_OF_CONDUCT.md # Contributor guidelines β”œβ”€β”€ CONTRIBUTING.md # Contribution guide └── SECURITY.md # Security reporting policies ```
## πŸ“Š Integrated Technologies & Licenses | Technology | License | Author / Maintainer | Source | |-----------------|---------------------|------------------------------------------|--------| | QualityScaler | MIT | [Djdefrag](https://github.com/Djdefrag) | [GitHub](https://github.com/Djdefrag/QualityScaler) | | RealScaler | MIT | [Djdefrag](https://github.com/Djdefrag) | [GitHub](https://github.com/Djdefrag/RealScaler) | | FluidFrames | MIT | [Djdefrag](https://github.com/Djdefrag) | [GitHub](https://github.com/Djdefrag/FluidFrames) | | Real-ESRGAN | BSD 3-Clause / Apache | Xintao Wang | [GitHub](https://github.com/xinntao/Real-ESRGAN) | | GFPGAN | Apache 2.0 | TencentARC / Xintao Wang | [GitHub](https://github.com/TencentARC/GFPGAN) | | RIFE | Apache 2.0 | hzwer | [GitHub](https://github.com/megvii-research/ECCV2022-RIFE) | | BSRGAN | Apache 2.0 | Kai Zhang | [GitHub](https://github.com/cszn/BSRGAN) | | IRCNN | BSD / Mixed | Kai Zhang | [GitHub](https://github.com/cszn/IRCNN) | | Anime4K | MIT | bloc97 | [GitHub](https://github.com/bloc97/Anime4K) | | ONNX Runtime | MIT | Microsoft | [GitHub](https://github.com/microsoft/onnxruntime) | | PyTorch | BSD 3-Clause | Meta AI | [GitHub](https://github.com/pytorch/pytorch) | | FFmpeg | LGPL / GPL | FFmpeg Team | [Official Site](https://ffmpeg.org) | | ExifTool | Artistic License | Phil Harvey | [Official Site](https://exiftool.org/) | | DirectML | MIT | Microsoft | [GitHub](https://github.com/microsoft/DirectML) | | Python | PSF License | Python Software Foundation | [Official Site](https://www.python.org) | | PyInstaller | GPLv2+ | PyInstaller Team | [GitHub](https://github.com/pyinstaller/pyinstaller) | | Inno Setup | Custom | Jordan Russell | [Official Site](http://www.jrsoftware.org/isinfo.php) | --- ## 🀝 Contributions We welcome contributions from the community: 1. **Fork** the repository. 2. **Create a branch** for your feature or bug fix. 3. **Submit a Pull Request** with detailed description and testing notes. πŸ“§ Contact: **[negroayub97@gmail.com](mailto:negroayub97@gmail.com)** --- ## πŸ“œ License Β© 2025 IvΓ‘n Eduardo Chavez Ayub Licensed under **MIT**. Additional terms and attributions are provided in `NOTICE.md`.