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/gaussian_yolo_layer.h | 44 ++++++++++++++++++++++---------------------- 1 files changed, 22 insertions(+), 22 deletions(-) diff --git a/lib/detecter_tools/darknet/gaussian_yolo_layer.h b/lib/detecter_tools/darknet/gaussian_yolo_layer.h index dac3929..9080881 100644 --- a/lib/detecter_tools/darknet/gaussian_yolo_layer.h +++ b/lib/detecter_tools/darknet/gaussian_yolo_layer.h @@ -1,22 +1,22 @@ -//Gaussian YOLOv3 implementation -#ifndef GAUSSIAN_YOLO_LAYER_H -#define GAUSSIAN_YOLO_LAYER_H - -#include "darknet.h" -#include "layer.h" -#include "network.h" - -layer make_gaussian_yolo_layer(int batch, int w, int h, int n, int total, int *mask, int classes, int max_boxes); -void forward_gaussian_yolo_layer(const layer l, network_state state); -void backward_gaussian_yolo_layer(const layer l, network_state state); -void resize_gaussian_yolo_layer(layer *l, int w, int h); -int gaussian_yolo_num_detections(layer l, float thresh); -int get_gaussian_yolo_detections(layer l, int w, int h, int netw, int neth, float thresh, int *map, int relative, detection *dets, int letter); -void correct_gaussian_yolo_boxes(detection *dets, int n, int w, int h, int netw, int neth, int relative, int letter); - -#ifdef GPU -void forward_gaussian_yolo_layer_gpu(const layer l, network_state state); -void backward_gaussian_yolo_layer_gpu(layer l, network_state state); -#endif - -#endif +//Gaussian YOLOv3 implementation +#ifndef GAUSSIAN_YOLO_LAYER_H +#define GAUSSIAN_YOLO_LAYER_H + +#include "darknet.h" +#include "layer.h" +#include "network.h" + +layer make_gaussian_yolo_layer(int batch, int w, int h, int n, int total, int *mask, int classes, int max_boxes); +void forward_gaussian_yolo_layer(const layer l, network_state state); +void backward_gaussian_yolo_layer(const layer l, network_state state); +void resize_gaussian_yolo_layer(layer *l, int w, int h); +int gaussian_yolo_num_detections(layer l, float thresh); +int get_gaussian_yolo_detections(layer l, int w, int h, int netw, int neth, float thresh, int *map, int relative, detection *dets, int letter); +void correct_gaussian_yolo_boxes(detection *dets, int n, int w, int h, int netw, int neth, int relative, int letter); + +#ifdef GPU +void forward_gaussian_yolo_layer_gpu(const layer l, network_state state); +void backward_gaussian_yolo_layer_gpu(layer l, network_state state); +#endif + +#endif -- Gitblit v1.8.0