| | |
| | | #ifndef IMAGE_H |
| | | #define IMAGE_H |
| | | |
| | | #include <stdlib.h> |
| | | #include <stdio.h> |
| | | #include <float.h> |
| | | #include <string.h> |
| | | #include <math.h> |
| | | #include "box.h" |
| | | |
| | | #ifdef OPENCV |
| | | #include "opencv2/highgui/highgui_c.h" |
| | | #include "opencv2/imgproc/imgproc_c.h" |
| | | #endif |
| | | |
| | | typedef struct { |
| | | int h; |
| | | int w; |
| | | int c; |
| | | double *data; |
| | | float *data; |
| | | } image; |
| | | |
| | | float get_color(int c, int x, int max); |
| | | void flip_image(image a); |
| | | void draw_box(image a, int x1, int y1, int x2, int y2, float r, float g, float b); |
| | | void draw_box_width(image a, int x1, int y1, int x2, int y2, int w, float r, float g, float b); |
| | | void draw_bbox(image a, box bbox, int w, float r, float g, float b); |
| | | void draw_label(image a, int r, int c, image label, const float *rgb); |
| | | void draw_detections(image im, int num, float thresh, box *boxes, float **probs, char **names, image *labels, int classes); |
| | | image image_distance(image a, image b); |
| | | void scale_image(image m, float s); |
| | | image crop_image(image im, int dx, int dy, int w, int h); |
| | | image random_crop_image(image im, int low, int high, int size); |
| | | image resize_image(image im, int w, int h); |
| | | image resize_min(image im, int min); |
| | | void translate_image(image m, float s); |
| | | void normalize_image(image p); |
| | | void threshold_image(image p, double t); |
| | | void zero_image(image m); |
| | | void rotate_image(image m); |
| | | image rotate_image(image m, float rad); |
| | | void rotate_image_cw(image im, int times); |
| | | void embed_image(image source, image dest, int dx, int dy); |
| | | void saturate_image(image im, float sat); |
| | | void exposure_image(image im, float sat); |
| | | void saturate_exposure_image(image im, float sat, float exposure); |
| | | void hsv_to_rgb(image im); |
| | | void rgbgr_image(image im); |
| | | void constrain_image(image im); |
| | | |
| | | void show_image(image p, char *name); |
| | | image grayscale_image(image im); |
| | | image threshold_image(image im, float thresh); |
| | | |
| | | image collapse_image_layers(image source, int border); |
| | | image collapse_images_horz(image *ims, int n); |
| | | image collapse_images_vert(image *ims, int n); |
| | | |
| | | void show_image(image p, const char *name); |
| | | void show_image_normalized(image im, const char *name); |
| | | void save_image(image p, const char *name); |
| | | void show_images(image *ims, int n, char *window); |
| | | void show_image_layers(image p, char *name); |
| | | void show_image_collapsed(image p, char *name); |
| | | |
| | | image make_image(int h, int w, int c); |
| | | image make_random_image(int h, int w, int c); |
| | | image make_random_kernel(int size, int c); |
| | | #ifdef OPENCV |
| | | void save_image_jpg(image p, char *name); |
| | | image get_image_from_stream(CvCapture *cap); |
| | | image ipl_to_image(IplImage* src); |
| | | #endif |
| | | |
| | | void print_image(image m); |
| | | |
| | | image make_image(int w, int h, int c); |
| | | image make_random_image(int w, int h, int c); |
| | | image make_empty_image(int w, int h, int c); |
| | | image float_to_image(int w, int h, int c, float *data); |
| | | image copy_image(image p); |
| | | image load_image(char *filename); |
| | | image load_image(char *filename, int w, int h, int c); |
| | | image load_image_color(char *filename, int w, int h); |
| | | |
| | | double get_pixel(image m, int x, int y, int c); |
| | | double get_pixel_extend(image m, int x, int y, int c); |
| | | void set_pixel(image m, int x, int y, int c, double val); |
| | | |
| | | float get_pixel(image m, int x, int y, int c); |
| | | float get_pixel_extend(image m, int x, int y, int c); |
| | | void set_pixel(image m, int x, int y, int c, float val); |
| | | void add_pixel(image m, int x, int y, int c, float val); |
| | | float bilinear_interpolate(image im, float x, float y, int c); |
| | | |
| | | image get_image_layer(image m, int l); |
| | | |
| | | void two_d_convolve(image m, int mc, image kernel, int kc, int stride, image out, int oc); |
| | | void upsample_image(image m, int stride, image out); |
| | | void convolve(image m, image kernel, int stride, int channel, image out); |
| | | void back_convolve(image m, image kernel, int stride, int channel, image out); |
| | | void kernel_update(image m, image update, int stride, int channel, image out); |
| | | |
| | | void free_image(image m); |
| | | void test_resize(char *filename); |
| | | #endif |
| | | |