#include "detection_layer.h"#include "activations.h"#include "softmax_layer.h"#include "blas.h"#include "box.h"#include "cuda.h"#include "utils.h"#include <stdio.h>#include <assert.h>#include <string.h>#include <stdlib.h>
Go to the source code of this file.
Functions | |
| void | backward_detection_layer (const detection_layer l, network_state state) |
| void | forward_detection_layer (const detection_layer l, network_state state) |
| void | get_detection_boxes (layer l, int w, int h, float thresh, float **probs, box *boxes, int only_objectness) |
| detection_layer | make_detection_layer (int batch, int inputs, int n, int side, int classes, int coords, int rescore) |
| void backward_detection_layer | ( | const detection_layer | l, |
| network_state | state | ||
| ) |
Definition at line 219 of file detection_layer.c.
| void forward_detection_layer | ( | const detection_layer | l, |
| network_state | state | ||
| ) |
Definition at line 49 of file detection_layer.c.
| void get_detection_boxes | ( | layer | l, |
| int | w, | ||
| int | h, | ||
| float | thresh, | ||
| float ** | probs, | ||
| box * | boxes, | ||
| int | only_objectness | ||
| ) |
Definition at line 224 of file detection_layer.c.
| detection_layer make_detection_layer | ( | int | batch, |
| int | inputs, | ||
| int | n, | ||
| int | side, | ||
| int | classes, | ||
| int | coords, | ||
| int | rescore | ||
| ) |
Definition at line 13 of file detection_layer.c.