yolo_layer.c File Reference

#include "yolo_layer.h"
#include "activations.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>

Include dependency graph for yolo_layer.c:

Go to the source code of this file.

Functions
layer	make_yolo_layer (int batch, int w, int h, int n, int total, int *mask, int classes)
void	resize_yolo_layer (layer *l, int w, int h)
box	get_yolo_box (float *x, float *biases, int n, int index, int i, int j, int lw, int lh, int w, int h, int stride)
float	delta_yolo_box (box truth, float *x, float *biases, int n, int index, int i, int j, int lw, int lh, int w, int h, float *delta, float scale, int stride)
void	delta_yolo_class (float *output, float *delta, int index, int class, int classes, int stride, float *avg_cat)
void	forward_yolo_layer (const layer l, network net)
void	backward_yolo_layer (const layer l, network net)
void	correct_yolo_boxes (detection *dets, int n, int w, int h, int netw, int neth, int relative)
int	yolo_num_detections (layer l, float thresh)
void	avg_flipped_yolo (layer l)
int	get_yolo_detections (layer l, int w, int h, int netw, int neth, float thresh, int *map, int relative, detection *dets)

Function Documentation

avg_flipped_yolo()

void avg_flipped_yolo

(

layer

l

)

Definition at line 290 of file yolo_layer.c.

backward_yolo_layer()

void backward_yolo_layer	(	const layer	l,
		network	net
	)

Definition at line 242 of file yolo_layer.c.

correct_yolo_boxes()

void correct_yolo_boxes	(	detection *	dets,
		int	n,
		int	w,
		int	h,
		int	netw,
		int	neth,
		int	relative
	)

Definition at line 247 of file yolo_layer.c.

delta_yolo_box()

float delta_yolo_box	(	box	truth,
		float *	x,
		float *	biases,
		int	n,
		int	index,
		int	i,
		int	j,
		int	lw,
		int	lh,
		int	w,
		int	h,
		float *	delta,
		float	scale,
		int	stride
	)

Definition at line 93 of file yolo_layer.c.

delta_yolo_class()

void delta_yolo_class	(	float *	output,
		float *	delta,
		int	index,
		int	class,
		int	classes,
		int	stride,
		float *	avg_cat
	)

Definition at line 111 of file yolo_layer.c.

forward_yolo_layer()

void forward_yolo_layer	(	const layer	l,
		network	net
	)

Definition at line 132 of file yolo_layer.c.

get_yolo_box()

box get_yolo_box	(	float *	x,
		float *	biases,
		int	n,
		int	index,
		int	i,
		int	j,
		int	lw,
		int	lh,
		int	w,
		int	h,
		int	stride
	)

Definition at line 83 of file yolo_layer.c.

get_yolo_detections()

int get_yolo_detections	(	layer	l,
		int	w,
		int	h,
		int	netw,
		int	neth,
		float	thresh,
		int *	map,
		int	relative,
		detection *	dets
	)

Definition at line 316 of file yolo_layer.c.

make_yolo_layer()

layer make_yolo_layer	(	int	batch,
		int	w,
		int	h,
		int	n,
		int	total,
		int *	mask,
		int	classes
	)

Definition at line 13 of file yolo_layer.c.

resize_yolo_layer()

void resize_yolo_layer	(	layer *	l,
		int	w,
		int	h
	)

Definition at line 63 of file yolo_layer.c.

yolo_num_detections()

int yolo_num_detections	(	layer	l,
		float	thresh
	)

Definition at line 275 of file yolo_layer.c.