darknet/src/box.h

23 lines
571 B
C
Raw Normal View History

2015-06-16 09:22:44 +03:00
#ifndef BOX_H
#define BOX_H
typedef struct{
float x, y, w, h;
} box;
typedef struct{
float dx, dy, dw, dh;
} dbox;
box float_to_box(float *f, int stride);
2015-06-16 09:22:44 +03:00
float box_iou(box a, box b);
2015-09-01 21:21:01 +03:00
float box_rmse(box a, box b);
2015-06-16 09:22:44 +03:00
dbox diou(box a, box b);
2015-09-01 21:21:01 +03:00
void do_nms(box *boxes, float **probs, int total, int classes, float thresh);
2015-11-04 06:23:17 +03:00
void do_nms_sort(box *boxes, float **probs, int total, int classes, float thresh);
2017-01-04 15:44:00 +03:00
void do_nms_obj(box *boxes, float **probs, int total, int classes, float thresh);
2015-08-25 04:27:42 +03:00
box decode_box(box b, box anchor);
box encode_box(box b, box anchor);
2015-06-16 09:22:44 +03:00
#endif