mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Added GIoU
This commit is contained in:
@ -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;
|
||||
|
Reference in New Issue
Block a user