Add the model card
Browse files
README.md
ADDED
|
@@ -0,0 +1,61 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: onnx
|
| 3 |
+
pipeline_tag: image-to-image
|
| 4 |
+
tags:
|
| 5 |
+
- super-resolution
|
| 6 |
+
- upscaling
|
| 7 |
+
- onnx
|
| 8 |
+
- image-upscaling
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# Upscaling models, as ONNX
|
| 12 |
+
|
| 13 |
+
The model files used by [Upscale Agent](https://github.com/benbaker76/UpscaleAgent), the
|
| 14 |
+
desktop app, and [UpsizerAI](https://github.com/benbaker76/UpsizerAI), the web app it is the
|
| 15 |
+
desktop version of. Both download from here on demand; nothing needs fetching by hand.
|
| 16 |
+
|
| 17 |
+
Each model is the original author's weights converted to ONNX by UpsizerAI's
|
| 18 |
+
`python/export_onnx.py`: opset and input/output naming are the same across all of them, and
|
| 19 |
+
height and width are dynamic so an image can be run whole or in tiles. Most come in two
|
| 20 |
+
precisions, `<name>.onnx` (float32 weights) and `<name>_fp16.onnx` (float16 weights, float32
|
| 21 |
+
input and output), which give the same picture to within one 8-bit level.
|
| 22 |
+
|
| 23 |
+
**Nothing here was trained by this organization.** Every file is a conversion, and each
|
| 24 |
+
remains under the license its author released the original weights under - follow the source
|
| 25 |
+
link for the terms, which for some models restrict commercial use.
|
| 26 |
+
|
| 27 |
+
| Model | Scale | fp32 | fp16 | Architecture | Original weights |
|
| 28 |
+
| --- | --- | --- | --- | --- | --- |
|
| 29 |
+
| `APISR_GRL_x4` | x4 | 27.1 MB | 17.5 MB | GRL | [HikariDawn/APISR](https://huggingface.co/HikariDawn/APISR) |
|
| 30 |
+
| `APISR_RRDB_x2` | x2 | 18.7 MB | 9.8 MB | RRDBNet-6B | [HikariDawn/APISR](https://huggingface.co/HikariDawn/APISR) |
|
| 31 |
+
| `AnimeSharpV2_ESRGAN_Soft_x2` | x2 | 70.0 MB | 36.6 MB | ESRGAN | [Kim2091/AnimeSharpV2](https://huggingface.co/Kim2091/AnimeSharpV2) |
|
| 32 |
+
| `AnimeSharpV2_MoSR_Sharp_x2` | x2 | 18.0 MB | 9.4 MB | MoSR | [Kim2091/AnimeSharpV2](https://huggingface.co/Kim2091/AnimeSharpV2) |
|
| 33 |
+
| `AnimeSharpV2_MoSR_Soft_x2` | x2 | 18.0 MB | 9.4 MB | MoSR | [Kim2091/AnimeSharpV2](https://huggingface.co/Kim2091/AnimeSharpV2) |
|
| 34 |
+
| `AnimeSharpV2_RPLKSR_Sharp_x2` | x2 | 30.8 MB | 16.1 MB | RealPLKSR | [Kim2091/AnimeSharpV2](https://huggingface.co/Kim2091/AnimeSharpV2) |
|
| 35 |
+
| `AnimeSharpV2_RPLKSR_Soft_x2` | x2 | 30.8 MB | 16.1 MB | RealPLKSR | [Kim2091/AnimeSharpV2](https://huggingface.co/Kim2091/AnimeSharpV2) |
|
| 36 |
+
| `AnimeSharpV3_x2` | x2 | 70.0 MB | 36.6 MB | ESRGAN | [Kim2091/AnimeSharpV3](https://huggingface.co/Kim2091/AnimeSharpV3) |
|
| 37 |
+
| `BSRGAN_x2` | x2 | 69.7 MB | 36.4 MB | RRDBNet | [kadirnar/BSRGANx2](https://huggingface.co/kadirnar/BSRGANx2) |
|
| 38 |
+
| `RealESRGAN_anime_x4` | x4 | 18.7 MB | 9.8 MB | RRDBNet-6B | [xinntao/Real-ESRGAN](https://github.com/xinntao/Real-ESRGAN) |
|
| 39 |
+
| `RealESRGAN_x2` | x2 | 70.0 MB | 36.6 MB | RRDBNet | [ai-forever/Real-ESRGAN](https://huggingface.co/ai-forever/Real-ESRGAN) |
|
| 40 |
+
| `RealESRGAN_x4` | x4 | 69.9 MB | 36.5 MB | RRDBNet | [ai-forever/Real-ESRGAN](https://huggingface.co/ai-forever/Real-ESRGAN) |
|
| 41 |
+
| `RealESRGAN_x4plus` | x4 | 69.9 MB | 36.5 MB | RRDBNet | [xinntao/Real-ESRGAN](https://github.com/xinntao/Real-ESRGAN) |
|
| 42 |
+
| `RealESRGAN_x8` | x8 | 70.0 MB | 36.6 MB | RRDBNet | [ai-forever/Real-ESRGAN](https://huggingface.co/ai-forever/Real-ESRGAN) |
|
| 43 |
+
| `RealESR_General_x4` | x4 | 5.0 MB | 2.5 MB | SRVGGNetCompact | [xinntao/Real-ESRGAN](https://github.com/xinntao/Real-ESRGAN) |
|
| 44 |
+
| `RealWebPhoto_RGT_x4` | x4 | 92.3 MB | - | RGT | [Phips/4xRealWebPhoto_RGT](https://huggingface.co/Phips/4xRealWebPhoto_RGT) |
|
| 45 |
+
| `Swin2SR_Classical_x2` | x2 | 72.4 MB | 40.0 MB | Swin2SR | [mv-lab/swin2sr](https://github.com/mv-lab/swin2sr) |
|
| 46 |
+
| `Swin2SR_Classical_x4` | x4 | 73.0 MB | 40.3 MB | Swin2SR | [mv-lab/swin2sr](https://github.com/mv-lab/swin2sr) |
|
| 47 |
+
| `Swin2SR_RealWorld_x4` | x4 | 72.3 MB | 39.9 MB | Swin2SR | [mv-lab/swin2sr](https://github.com/mv-lab/swin2sr) |
|
| 48 |
+
| `SwinIR_BSRGAN_x4` | x4 | 55.2 MB | - | SwinIR-M | [mikestealth/SwinIR](https://huggingface.co/mikestealth/SwinIR) |
|
| 49 |
+
| `UltraMix_Smooth_x4` | x4 | 69.9 MB | 36.5 MB | ESRGAN | [Kim2091/UltraSharp](https://huggingface.co/Kim2091/UltraSharp) |
|
| 50 |
+
| `UltraSharp_x4` | x4 | 69.9 MB | 36.5 MB | ESRGAN | [Kim2091/UltraSharp](https://huggingface.co/Kim2091/UltraSharp) |
|
| 51 |
+
|
| 52 |
+
## Using a file directly
|
| 53 |
+
|
| 54 |
+
```
|
| 55 |
+
https://huggingface.co/baker76/upscale-models/resolve/main/RealESRGAN_x4.onnx
|
| 56 |
+
```
|
| 57 |
+
|
| 58 |
+
Input is `[1, 3, H, W]` float32, RGB, 0 to 1; output is the same layout at the model's
|
| 59 |
+
scale. The x2 RRDBNet models need even sides, and the window-attention models need sides
|
| 60 |
+
that are a multiple of 8 or 32; `models.json` in the UpsizerAI repository records the
|
| 61 |
+
multiple, and the peak activation, of each file.
|