Add model card
Browse files
README.md
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
pipeline_tag: image-to-image
|
| 4 |
+
library_name: coreai
|
| 5 |
+
tags:
|
| 6 |
+
- core-ai
|
| 7 |
+
- ios
|
| 8 |
+
- super-resolution
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# EDSR x2 for Core AI
|
| 12 |
+
|
| 13 |
+
EDSR (`edsr_r16f64_x2`, 1.5M parameters) from [torchSR](https://github.com/Coloquinte/torchSR), exported to Apple Core AI with the [`apple/coreai-models`](https://github.com/apple/coreai-models/tree/main/models/edsr) recipe:
|
| 14 |
+
|
| 15 |
+
```sh
|
| 16 |
+
uv run models/edsr/export.py --dtype float16 --dynamic
|
| 17 |
+
```
|
| 18 |
+
|
| 19 |
+
| Folder | Contents | Size | Platforms |
|
| 20 |
+
| ----------------- | -------------------------------------------------------------- | ------ | ---------- |
|
| 21 |
+
| `edsr-x2.aimodel` | `main(x)`: `x` is `[batch, 3, height, width]` RGB in 0...1, float16, batch 1–64, height and width 8–256. Returns the image at twice the size | 2.6 MB | iOS, macOS |
|
| 22 |
+
|
| 23 |
+
Images larger than 256 pixels are upscaled in tiles.
|
| 24 |
+
|
| 25 |
+
## Use
|
| 26 |
+
|
| 27 |
+
With [`@react-native-ai/core-ai`](https://github.com/callstackincubator/ai/tree/main/packages/core-ai), download `edsr-x2.aimodel` into app storage, then call it with `CoreAI.runRawFunction`, or through `CoreAI.runTask('super-resolution', ...)`, which tiles and converts images for you.
|
| 28 |
+
|
| 29 |
+
## License
|
| 30 |
+
|
| 31 |
+
torchSR and its EDSR weights are MIT licensed ([EDSR paper](https://arxiv.org/abs/1707.02921)). This repository only converts them to Core AI format.
|