metadata
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, exported to Apple Core AI with the apple/coreai-models recipe:
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, 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). This repository only converts them to Core AI format.