From 168af40fe9a3cc81c6ee16b3e81f154780c36bdb Mon Sep 17 00:00:00 2001 From: Scheaven <xuepengqiang> Date: 星期四, 03 六月 2021 15:03:27 +0800 Subject: [PATCH] up new v4 --- lib/detecter_tools/darknet/region_layer.h | 58 +++++++++++++++++++++++++++++----------------------------- 1 files changed, 29 insertions(+), 29 deletions(-) diff --git a/lib/detecter_tools/darknet/region_layer.h b/lib/detecter_tools/darknet/region_layer.h index f95755b..e616624 100644 --- a/lib/detecter_tools/darknet/region_layer.h +++ b/lib/detecter_tools/darknet/region_layer.h @@ -1,29 +1,29 @@ -#ifndef REGION_LAYER_H -#define REGION_LAYER_H - -#include "layer.h" -#include "network.h" - -typedef layer region_layer; - -#ifdef __cplusplus -extern "C" { -#endif -region_layer make_region_layer(int batch, int w, int h, int n, int classes, int coords, int max_boxes); -void forward_region_layer(const region_layer l, network_state state); -void backward_region_layer(const region_layer l, network_state state); -void get_region_boxes(layer l, int w, int h, float thresh, float **probs, box *boxes, int only_objectness, int *map); -void resize_region_layer(layer *l, int w, int h); -void get_region_detections(layer l, int w, int h, int netw, int neth, float thresh, int *map, float tree_thresh, int relative, detection *dets); -void correct_region_boxes(detection *dets, int n, int w, int h, int netw, int neth, int relative); -void zero_objectness(layer l); - -#ifdef GPU -void forward_region_layer_gpu(const region_layer l, network_state state); -void backward_region_layer_gpu(region_layer l, network_state state); -#endif - -#ifdef __cplusplus -} -#endif -#endif +#ifndef REGION_LAYER_H +#define REGION_LAYER_H + +#include "layer.h" +#include "network.h" + +typedef layer region_layer; + +#ifdef __cplusplus +extern "C" { +#endif +region_layer make_region_layer(int batch, int w, int h, int n, int classes, int coords, int max_boxes); +void forward_region_layer(const region_layer l, network_state state); +void backward_region_layer(const region_layer l, network_state state); +void get_region_boxes(layer l, int w, int h, float thresh, float **probs, box *boxes, int only_objectness, int *map); +void resize_region_layer(layer *l, int w, int h); +void get_region_detections(layer l, int w, int h, int netw, int neth, float thresh, int *map, float tree_thresh, int relative, detection *dets); +void correct_region_boxes(detection *dets, int n, int w, int h, int netw, int neth, int relative); +void zero_objectness(layer l); + +#ifdef GPU +void forward_region_layer_gpu(const region_layer l, network_state state); +void backward_region_layer_gpu(region_layer l, network_state state); +#endif + +#ifdef __cplusplus +} +#endif +#endif -- Gitblit v1.8.0