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.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Paper for grabbou/edsr-x2-CoreAI