File size: 906 Bytes
e4545e2 |
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 33 34 35 36 37 38 39 |
#include <opencv2/core.hpp>
#include <cstdlib>
#include <cstdio>
#include <cstring>
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" */
|