Spaces:
Running
on
A10G
Running
on
A10G
| extern "C" { | |
| struct PM_shape_t { | |
| int width, height, channels; | |
| }; | |
| enum PM_dtype_e { | |
| PM_UINT8, | |
| PM_INT8, | |
| PM_UINT16, | |
| PM_INT16, | |
| PM_INT32, | |
| PM_FLOAT32, | |
| PM_FLOAT64, | |
| }; | |
| struct PM_mat_t { | |
| void *data_ptr; | |
| PM_shape_t shape; | |
| int dtype; | |
| }; | |
| void PM_set_random_seed(unsigned int seed); | |
| void PM_set_verbose(int value); | |
| void PM_free_pymat(PM_mat_t pymat); | |
| PM_mat_t PM_inpaint(PM_mat_t image, PM_mat_t mask, int patch_size); | |
| PM_mat_t PM_inpaint_regularity(PM_mat_t image, PM_mat_t mask, PM_mat_t ijmap, int patch_size, float guide_weight); | |
| PM_mat_t PM_inpaint2(PM_mat_t image, PM_mat_t mask, PM_mat_t global_mask, int patch_size); | |
| PM_mat_t PM_inpaint2_regularity(PM_mat_t image, PM_mat_t mask, PM_mat_t global_mask, PM_mat_t ijmap, int patch_size, float guide_weight); | |
| } /* extern "C" */ | |