luboslenco
51 lines
1.3 KiB
C
51 lines
1.3 KiB
C
/*
|
|
* iris_upscale.h - RealESRGAN_x4plus image upscaler (RRDBNet)
|
|
*
|
|
* Self-contained, CPU-only inference for the RealESRGAN_x4plus model
|
|
* (RRDBNet: scale=4, num_feat=64, num_block=23, num_grow_ch=32).
|
|
* Loads weights from a single safetensors file and upscales an RGB image
|
|
* by 4x.
|
|
*/
|
|
|
|
#ifndef IRIS_UPSCALE_H
|
|
#define IRIS_UPSCALE_H
|
|
|
|
#include "iris.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
typedef struct iris_upscale iris_upscale_t;
|
|
|
|
/*
|
|
* Load the RealESRGAN_x4plus model from a safetensors file.
|
|
* Returns NULL on error (use iris_get_error() for details).
|
|
*/
|
|
iris_upscale_t *iris_upscale_load(const char *path);
|
|
|
|
/*
|
|
* Enable "tileable" mode: subtract the smoothest periodic field that makes the
|
|
* upscaled image wrap seamlessly, so a tileable input stays tileable after 4x
|
|
* upscaling. Off by default.
|
|
*/
|
|
void iris_upscale_set_tileable(iris_upscale_t *model, int on);
|
|
|
|
/*
|
|
* Free the model and its resources.
|
|
*/
|
|
void iris_upscale_free(iris_upscale_t *model);
|
|
|
|
/*
|
|
* Upscale an RGB(A) image by 4x. Alpha is ignored; the result is always
|
|
* a 3-channel RGB image with dimensions (4*width, 4*height).
|
|
* Returns a newly allocated image (free with iris_image_free), NULL on error.
|
|
*/
|
|
iris_image *iris_upscale_run(iris_upscale_t *model, const iris_image *input);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif /* IRIS_UPSCALE_H */
|