From c62b4f35aa2c59d7db0fd177affeed14b1ba4bcb Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 08 Sep 2016 07:04:39 +0000
Subject: [PATCH] adding coco models
---
src/layer.h | 16 ++++++++++++++++
1 files changed, 16 insertions(+), 0 deletions(-)
diff --git a/src/layer.h b/src/layer.h
index d53fe38..5e9dc3a 100644
--- a/src/layer.h
+++ b/src/layer.h
@@ -28,6 +28,9 @@
CRNN,
BATCHNORM,
NETWORK,
+ XNOR,
+ REGION,
+ REORG,
BLANK
} LAYER_TYPE;
@@ -69,6 +72,7 @@
float saturation;
float exposure;
float shift;
+ float ratio;
int softmax;
int classes;
int coords;
@@ -78,6 +82,8 @@
int does_cost;
int joint;
int noadjust;
+ int reorg;
+ int log;
float alpha;
float beta;
@@ -87,6 +93,7 @@
float object_scale;
float noobject_scale;
float class_scale;
+ int random;
int dontload;
int dontloadscales;
@@ -102,6 +109,9 @@
char *cfilters;
float *filter_updates;
float *state;
+ float *prev_state;
+ float *forgot_state;
+ float *forgot_delta;
float *state_delta;
float *concat;
@@ -159,6 +169,12 @@
struct layer *input_h_layer;
struct layer *state_h_layer;
+ float *z_cpu;
+ float *r_cpu;
+ float *h_cpu;
+
+ float *binary_input;
+
size_t workspace_size;
#ifdef GPU
--
Gitblit v1.10.0