Go to the documentation of this file.00001 #ifndef BOX_H
00002 #define BOX_H
00003
00004 typedef struct{
00005 float x, y, w, h;
00006 } box;
00007
00008 typedef struct{
00009 float dx, dy, dw, dh;
00010 } dbox;
00011
00012 box float_to_box(float *f);
00013 float box_iou(box a, box b);
00014 float box_rmse(box a, box b);
00015 dbox diou(box a, box b);
00016 void do_nms(box *boxes, float **probs, int total, int classes, float thresh);
00017 void do_nms_sort(box *boxes, float **probs, int total, int classes, float thresh);
00018 box decode_box(box b, box anchor);
00019 box encode_box(box b, box anchor);
00020
00021 #endif