![Warlock-Studio banner](Assets/banner.png)
### _AI Media Enhancement Suite_ [![Build Status](https://img.shields.io/badge/Build-Stable_Release-0A192F?style=for-the-badge&logo=github&logoColor=FFD700)](https://github.com/Ivan-Ayub97/Warlock-Studio/releases) [![Version](https://img.shields.io/badge/Version-4.2-FF8C00?style=for-the-badge&logo=git&logoColor=white)](https://github.com/Ivan-Ayub97/Warlock-Studio/releases/tag/4.2) [![License](https://img.shields.io/badge/License-MIT-6A0DAD?style=for-the-badge&logo=open-source-initiative&logoColor=white)](LICENSE) [![Downloads](https://img.shields.io/github/downloads/Ivan-Ayub97/Warlock-Studio/total?style=for-the-badge&color=FFD700&logo=download&logoColor=black)](https://github.com/Ivan-Ayub97/Warlock-Studio/releases) [![Platform](https://img.shields.io/badge/Platform-Windows-0078D6?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=FFD700)](https://www.python.org/) [![Issues](https://img.shields.io/github/issues/Ivan-Ayub97/Warlock-Studio?style=for-the-badge&color=FF4500&logo=github&logoColor=white)](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&color=2E8B57&logo=git&logoColor=white)](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.2** introduces a **full offline installer**, an **advanced ONNX Runtime engine** with CUDA support, and significant **packaging optimizations**, ensuring the most reliable and performant experience yet. --- ## πŸ“₯ Download Installer (v4.2) 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**. - **Advanced Hardware Acceleration** – Intelligent provider selection prioritizes **CUDA**, falls back to **DirectML**, and finally **CPU** for maximum compatibility and performance. - **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.2 - πŸ“¦ **Full Offline Installer**: The application is now distributed as a single, self-contained offline installer. All AI models are included, eliminating the need for an internet connection during setup and ensuring a reliable installation. - πŸš€ **Advanced ONNX Runtime Engine**: The model loading architecture was re-engineered to intelligently prioritize hardware acceleration providers (**CUDA** > **DirectML** > **CPU**), maximizing performance on capable hardware and ensuring stability via a robust fallback mechanism. - βš™οΈ **Aggressive Packaging Optimization**: The final application size has been drastically reduced by aggressively pruning unnecessary dependencies from the PyInstaller build, resulting in a lighter and more efficient package. - πŸ› **Enhanced Runtime Stability**: Added crucial hidden imports to the build process, preventing `ModuleNotFoundError` crashes and ensuring all components of `onnxruntime` and other libraries load correctly. - πŸ–₯️ **Improved Debugging Experience**: The application now runs with an attached console window, providing real-time logs and error messages for easier troubleshooting. - ✨ **Professional Splash Screen**: A new startup splash screen provides immediate visual feedback while the application initializes, improving the user experience. --- ## πŸ–ΌοΈ 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 the Full Offline Installer** (see links above). 2. Run the **setup wizard** and follow the prompts. 3. Launch via Start Menu or Desktop shortcut. > Warlock-Studio is packaged using **PyInstaller** and deployed with **Inno Setup** for a seamless, self-contained installation experience. ### Installer Previews ![Installer 1](rsc/Installation_window.png) ![Installer 2](rsc/Installation_window2.png) ![Installer 3](rsc/Installation_window3.png) --- ## πŸ–₯️ System Requirements - **OS:** Windows 11 or higher (64-bit) - **RAM:** 8GB+ recommended - **GPU:** NVIDIA (for CUDA), AMD, or Intel GPU with up-to-date drivers recommended - **Storage:** Sufficient free space for input and processed media --- ## πŸ“Œ Development Status (v4.2) | Component | Status | Notes | | ---------------------------- | ----------- | -------------------------------------------------------- | | ONNX Runtime Engine | 🟒 Enhanced | Prioritizes CUDA > DirectML > CPU with automatic fallback. | | Installer & Packaging | 🟒 Overhauled | Full offline installer; heavily optimized package size. | | Upscaling Models | 🟒 Stable | Includes VRAM recovery integration. | | 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 splash screen. | | Code Quality | 🟒 Improved | Refactored, modular, and more maintainable. | --- ## πŸ“‚ 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 β”œβ”€β”€ 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) | | 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) | | ONNX Runtime | MIT | Microsoft | [GitHub](https://github.com/microsoft/onnxruntime) | | FFmpeg | LGPL / GPL | FFmpeg Team | [Official Site](https://ffmpeg.org) | | ExifTool | Artistic License | Phil Harvey | [Official Site](https://exiftool.org/) | | 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 a 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`.