| | --- |
| | title: Color Restorization Model |
| | emoji: 🖼️ |
| | colorFrom: indigo |
| | colorTo: yellow |
| | sdk: gradio |
| | sdk_version: 3.9 |
| | app_file: app.py |
| | pinned: true |
| | license: unlicense |
| | --- |
| | |
| | # 🌈 Color Restorization Model (CPU Optimized) |
| |
|
| | Bring your old black & white photos back to life—upload, adjust, and download in vivid color. |
| |
|
| | This version has been optimized for **CPU inference**, removing GPU dependencies and improving performance on standard hardware. |
| |
|
| | ## Features |
| |
|
| | * **Adaptive Resolution Processing**: Large images are processed intelligently to preserve sharpness while ensuring fast colorization. |
| | * **Quality Presets**: Choose between **Fast**, **Balanced**, and **High** quality to suit your hardware. |
| | * **Real-time Progress**: Visual progress bar. |
| | * **Pure CPU Stack**: Optimized for Intel/AMD CPUs with AVX2 support (via PyTorch). |
| |
|
| | ## CPU Compatibility Matrix |
| |
|
| | | Processor Generation | Recommended Preset | 1080p Processing Time (Est.) | |
| | | :--- | :--- | :--- | |
| | | Intel Core i3 / Older | **Fast (256px)** | 2-5s | |
| | | Intel Core i5 (8th Gen+) | **Balanced (512px)** | 1-3s | |
| | | Intel Core i7 / Ryzen 7 | **High (1080px)** | 3-8s | |
| | | M1/M2 Mac | **Balanced** | <1s | |
| |
|
| | ## Performance Tuning |
| |
|
| | * **Memory Constrained (<8GB RAM):** Stick to "Fast" or "Balanced". |
| | * **High-Res Archival:** Use "Original" resolution only if you have >16GB RAM and patience. |
| | * **Batch Processing:** The core logic is thread-safe and can be extended for batch processing. |
| |
|
| | ## Technical Details |
| |
|
| | The application uses the DDColor architecture via ModelScope. Optimizations include: |
| | 1. **L-Channel Preservation:** We apply colorization at a lower resolution and merge it with the original high-resolution Luminance channel using LAB color space. |
| | 2. **In-Memory Pipeline:** Removed disk I/O bottlenecks. |
| | 3. **Dynamic Quantization:** Automatically applied to the model on supported CPUs. |
| |
|
| | ## Installation |
| |
|
| | ```bash |
| | pip install -r requirements.txt |
| | python app.py |
| | ``` |
| |
|
| | Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference |
| |
|
