| | |
|---|
| | | #include "route_layer.h" |
|---|
| | | #include "shortcut_layer.h" |
|---|
| | | #include "yolo_layer.h" |
|---|
| | | #include "upsample_layer.h" |
|---|
| | | #include "parser.h" |
|---|
| | | |
|---|
| | | network *load_network_custom(char *cfg, char *weights, int clear, int batch) |
|---|
| | |
|---|
| | | box *boxes = calloc(l.w*l.h*l.n, sizeof(box)); |
|---|
| | | float **probs = calloc(l.w*l.h*l.n, sizeof(float *)); |
|---|
| | | int i, j; |
|---|
| | | for (j = 0; j < l.w*l.h*l.n; ++j) probs[j] = calloc(l.classes, sizeof(float *)); |
|---|
| | | for (j = 0; j < l.w*l.h*l.n; ++j) probs[j] = calloc(l.classes, sizeof(float)); |
|---|
| | | get_region_boxes(l, 1, 1, thresh, probs, boxes, 0, map); |
|---|
| | | for (j = 0; j < l.w*l.h*l.n; ++j) { |
|---|
| | | dets[j].classes = l.classes; |
|---|
