From 989ab8c38a02fa7ea9c25108151736c62e81c972 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 24 Apr 2015 17:27:50 +0000
Subject: [PATCH] IOU loss function
---
src/image.h | 48 ++++++++++++++++++++++++++++--------------------
1 files changed, 28 insertions(+), 20 deletions(-)
diff --git a/src/image.h b/src/image.h
index 0d7d6e2..0cffc7b 100644
--- a/src/image.h
+++ b/src/image.h
@@ -1,6 +1,7 @@
#ifndef IMAGE_H
#define IMAGE_H
+
#include "opencv2/highgui/highgui_c.h"
#include "opencv2/imgproc/imgproc_c.h"
typedef struct {
@@ -10,44 +11,51 @@
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);
+image image_distance(image a, image b);
void scale_image(image m, float s);
-void add_scalar_image(image m, float s);
+image crop_image(image im, int dx, int dy, int w, int h);
+image resize_image(image im, int w, int h);
+void translate_image(image m, float s);
void normalize_image(image p);
-void z_normalize_image(image p);
-void threshold_image(image p, float t);
-void zero_image(image m);
-void rotate_image(image m);
-void subtract_image(image a, image b);
-float avg_image_layer(image m, int l);
-void embed_image(image source, image dest, int h, int w);
+image rotate_image(image m, float rad);
+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);
+
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, char *name);
+void save_image(image p, 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);
+
void print_image(image m);
-image make_image(int h, int w, int c);
-image make_empty_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, float scale);
-image float_to_image(int h, int w, int c, float *data);
+image make_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, int h, int w);
+image load_image(char *filename, int w, int h);
+image load_image_color(char *filename, int w, int h);
+
image ipl_to_image(IplImage* src);
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);
+float billinear_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, int edge);
-void upsample_image(image m, int stride, image out);
-void convolve(image m, image kernel, int stride, int channel, image out, int edge);
-void back_convolve(image m, image kernel, int stride, int channel, image out, int edge);
-void kernel_update(image m, image update, int stride, int channel, image out, int edge);
-
void free_image(image m);
+void test_resize(char *filename);
#endif
--
Gitblit v1.10.0