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/col2im.h | 66 ++++++++++++++++---------------- 1 files changed, 33 insertions(+), 33 deletions(-) diff --git a/lib/detecter_tools/darknet/col2im.h b/lib/detecter_tools/darknet/col2im.h index 3bd8718..984f7c4 100644 --- a/lib/detecter_tools/darknet/col2im.h +++ b/lib/detecter_tools/darknet/col2im.h @@ -1,33 +1,33 @@ -#ifndef COL2IM_H -#define COL2IM_H - -#ifdef __cplusplus -extern "C" { -#endif -void col2im_cpu(float* data_col, - int channels, int height, int width, - int ksize, int stride, int pad, float* data_im); - -void col2im_cpu_ext(const float* data_col, const int channels, - const int height, const int width, const int kernel_h, const int kernel_w, - const int pad_h, const int pad_w, - const int stride_h, const int stride_w, - const int dilation_h, const int dilation_w, - float* data_im); - -#ifdef GPU -void col2im_ongpu(float *data_col, - int channels, int height, int width, - int ksize, int stride, int pad, float *data_im); - - -void col2im_gpu_ext(const float* data_col, const int channels, - const int height, const int width, const int kernel_h, const int kernel_w, - const int pad_h, const int pad_w, const int stride_h, - const int stride_w, const int dilation_h, const int dilation_w, - float* data_im); -#endif -#ifdef __cplusplus -} -#endif -#endif +#ifndef COL2IM_H +#define COL2IM_H + +#ifdef __cplusplus +extern "C" { +#endif +void col2im_cpu(float* data_col, + int channels, int height, int width, + int ksize, int stride, int pad, float* data_im); + +void col2im_cpu_ext(const float* data_col, const int channels, + const int height, const int width, const int kernel_h, const int kernel_w, + const int pad_h, const int pad_w, + const int stride_h, const int stride_w, + const int dilation_h, const int dilation_w, + float* data_im); + +#ifdef GPU +void col2im_ongpu(float *data_col, + int channels, int height, int width, + int ksize, int stride, int pad, float *data_im); + + +void col2im_gpu_ext(const float* data_col, const int channels, + const int height, const int width, const int kernel_h, const int kernel_w, + const int pad_h, const int pad_w, const int stride_h, + const int stride_w, const int dilation_h, const int dilation_w, + float* data_im); +#endif +#ifdef __cplusplus +} +#endif +#endif -- Gitblit v1.8.0