From fe6e694e177b7c85d9a54734d2ba602b03159536 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 13 Sep 2017 10:47:19 +0000
Subject: [PATCH] Fixed yolo_console_dll.cpp
---
src/normalization_layer.h | 29 +++++++++++------------------
1 files changed, 11 insertions(+), 18 deletions(-)
diff --git a/src/normalization_layer.h b/src/normalization_layer.h
index 914fe7d..ab32776 100644
--- a/src/normalization_layer.h
+++ b/src/normalization_layer.h
@@ -2,25 +2,18 @@
#define NORMALIZATION_LAYER_H
#include "image.h"
+#include "layer.h"
+#include "network.h"
-typedef struct {
- int batch;
- int h,w,c;
- int size;
- float alpha;
- float beta;
- float kappa;
- float *delta;
- float *output;
- float *sums;
-} normalization_layer;
+layer make_normalization_layer(int batch, int w, int h, int c, int size, float alpha, float beta, float kappa);
+void resize_normalization_layer(layer *layer, int h, int w);
+void forward_normalization_layer(const layer layer, network_state state);
+void backward_normalization_layer(const layer layer, network_state state);
+void visualize_normalization_layer(layer layer, char *window);
-image get_normalization_image(normalization_layer layer);
-normalization_layer *make_normalization_layer(int batch, int h, int w, int c, int size, float alpha, float beta, float kappa);
-void resize_normalization_layer(normalization_layer *layer, int h, int w);
-void forward_normalization_layer(const normalization_layer layer, float *in);
-void backward_normalization_layer(const normalization_layer layer, float *in, float *delta);
-void visualize_normalization_layer(normalization_layer layer, char *window);
-
+#ifdef GPU
+void forward_normalization_layer_gpu(const layer layer, network_state state);
+void backward_normalization_layer_gpu(const layer layer, network_state state);
#endif
+#endif
--
Gitblit v1.10.0