Refactor processing of image, Add alpha channel

[colorize.git] / src / opticalflow.h

diff --git a/src/opticalflow.h b/src/opticalflow.h
index a144d1f..59be9eb 100644 (file)
--- a/src/opticalflow.h
+++ b/src/opticalflow.h
@@ -4,6 +4,6 @@ void flow_default(void);
 void save_flow(const char *filename);
 int load_flow(const char *filename);
 
-void *create_flow_maps(const unsigned short *img_prev_buffer, const unsigned short *img_next_buffer, unsigned short *img_buffer, int width, int height, int win_size, int steps, double *flow_map_x_prev, double *flow_map_y_prev, double *flow_map_x_next, double *flow_map_y_next, void *_image_preview);
+void *create_flow_maps(const double *img_prev_buffer, const double *img_next_buffer, const double *img_buffer, int width, int height, int win_size, int steps, double *flow_map_x_prev, double *flow_map_y_prev, double *flow_map_x_next, double *flow_map_y_next, void *_image_preview);
 
 void create_flow_view(const char *filename_prev, const char *filename_next, unsigned short *img_buffer, int width, int height, int win_size, int steps, int uv_scale);