Phips commited on
Commit
a1be0f2
·
verified ·
1 Parent(s): a1b7f41

add BODY suite + relation to NERVE

Browse files
Files changed (1) hide show
  1. README.md +20 -0
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).