File size: 1,356 Bytes
abf97c6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | ---
license: mit
pipeline_tag: image-to-image
library_name: coreai
tags:
- core-ai
- ios
- super-resolution
---
# EDSR x2 for Core AI
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:
```sh
uv run models/edsr/export.py --dtype float16 --dynamic
```
| Folder | Contents | Size | Platforms |
| ----------------- | -------------------------------------------------------------- | ------ | ---------- |
| `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 |
Images larger than 256 pixels are upscaled in tiles.
## Use
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.
## License
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.
|