File size: 906 Bytes
09c675d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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" */