Added GIoU

This commit is contained in:
AlexeyAB
2019-06-01 01:27:18 +03:00
parent 55dcd1bcb8
commit 6e13527f06
13 changed files with 3551 additions and 31 deletions

View File

@ -105,6 +105,11 @@ typedef enum {
LOGISTIC, RELU, RELIE, LINEAR, RAMP, TANH, PLSE, LEAKY, ELU, LOGGY, STAIR, HARDTAN, LHTAN, SELU
}ACTIVATION;
// parser.h
typedef enum {
IOU, GIOU, MSE
} IOU_LOSS;
// image.h
typedef enum{
PNG, BMP, TGA, JPG
@ -309,6 +314,10 @@ struct layer {
float *weights;
float *weight_updates;
float iou_normalizer;
float cls_normalizer;
int iou_loss;
char *align_bit_weights_gpu;
float *mean_arr_gpu;
float *align_workspace_gpu;
@ -671,6 +680,24 @@ typedef struct box {
float x, y, w, h;
} box;
// box.h
typedef struct boxabs {
float left, right, top, bot;
} boxabs;
// box.h
typedef struct dxrep {
float dt, db, dl, dr;
} dxrep;
// box.h
typedef struct ious {
float iou, giou;
dxrep dx_iou;
dxrep dx_giou;
} ious;
// box.h
typedef struct detection{
box bbox;