Image-to-Image
PyTorch
ONNX
TensorRT
English
super-resolution
image-restoration
sisr
real-world-restoration
spandrel
chainner
transformer
attention
Instructions to use Phips/HEART with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- TensorRT
How to use Phips/HEART with TensorRT:
# No code snippets available yet for this library. # To use this model, check the repository files and the library's documentation. # Want to help? PRs adding snippets are welcome at: # https://github.com/huggingface/huggingface.js
- Notebooks
- Google Colab
- Kaggle
add BODY suite + relation to NERVE
Browse files
README.md
CHANGED
|
@@ -20,6 +20,10 @@ A super-resolution architecture that is simple to train, fast to run, and
|
|
| 20 |
works everywhere: **traiNNer-redux**, **chaiNNer**, **ONNX Runtime**,
|
| 21 |
**TensorRT**.
|
| 22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
> Short version: copy `heart_arch.py` into traiNNer-redux, train with
|
| 24 |
> `lq_size: 96`, export to ONNX, use anywhere. Details below.
|
| 25 |
|
|
@@ -123,6 +127,22 @@ scripts/export_heart.py checkpoint → dynamic ONNX exporter
|
|
| 123 |
`examples/inputs/` also contains the original Real-ESRGAN test images, so you
|
| 124 |
can inspect full-size inputs and run the models yourself.
|
| 125 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 126 |
## License & credits
|
| 127 |
|
| 128 |
Apache-2.0 (this repository's code and pretrained weights).
|
|
|
|
| 20 |
works everywhere: **traiNNer-redux**, **chaiNNer**, **ONNX Runtime**,
|
| 21 |
**TensorRT**.
|
| 22 |
|
| 23 |
+
Part of the **BODY** suite by **Philip Hofmann** — networks built to be
|
| 24 |
+
real-world usable first, not benchmark-chasing. See
|
| 25 |
+
[Relation to NERVE](#relation-to-nerve) for which one to pick.
|
| 26 |
+
|
| 27 |
> Short version: copy `heart_arch.py` into traiNNer-redux, train with
|
| 28 |
> `lq_size: 96`, export to ONNX, use anywhere. Details below.
|
| 29 |
|
|
|
|
| 127 |
`examples/inputs/` also contains the original Real-ESRGAN test images, so you
|
| 128 |
can inspect full-size inputs and run the models yourself.
|
| 129 |
|
| 130 |
+
## Relation to NERVE
|
| 131 |
+
|
| 132 |
+
HEART and NERVE are the two members of the **BODY** suite (by Philip Hofmann):
|
| 133 |
+
same philosophy — real-world usability over benchmark scores — different tiers.
|
| 134 |
+
|
| 135 |
+
| | **HEART** | **NERVE** |
|
| 136 |
+
|---|---|---|
|
| 137 |
+
| size | ~16.7M params | ~1.8M params |
|
| 138 |
+
| design | attention (FlashAttention-friendly) | pure convolution |
|
| 139 |
+
| use when | you want the **best quality** and have the compute | you want **speed, small size, and dead-simple deployment** (mobile/edge/web) |
|
| 140 |
+
| cost | heavier to run | a fraction of the cost |
|
| 141 |
+
|
| 142 |
+
Rule of thumb: **HEART for maximum quality, NERVE for maximum practicality.**
|
| 143 |
+
Both export to dynamic ONNX; NERVE additionally converts to ncnn cleanly
|
| 144 |
+
because it has no attention ops.
|
| 145 |
+
|
| 146 |
## License & credits
|
| 147 |
|
| 148 |
Apache-2.0 (this repository's code and pretrained weights).
|