#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.