From 2a1025b6f0cb2bf54f771d9c56851ed66336e1a0 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 22 Sep 2015 23:40:51 +0000
Subject: [PATCH] save things sometimes

---
 src/layer.h |   14 ++++++++++++++
 1 files changed, 14 insertions(+), 0 deletions(-)

diff --git a/src/layer.h b/src/layer.h
index c319183..d13cdbf 100644
--- a/src/layer.h
+++ b/src/layer.h
@@ -15,6 +15,7 @@
     ROUTE,
     COST,
     NORMALIZATION,
+    REGION,
     AVGPOOL
 } LAYER_TYPE;
 
@@ -29,19 +30,23 @@
     int batch;
     int inputs;
     int outputs;
+    int truths;
     int h,w,c;
     int out_h, out_w, out_c;
     int n;
     int groups;
     int size;
+    int side;
     int stride;
     int pad;
     int crop_width;
     int crop_height;
+    int sqrt;
     int flip;
     float angle;
     float saturation;
     float exposure;
+    int softmax;
     int classes;
     int coords;
     int background;
@@ -49,15 +54,22 @@
     int objectness;
     int does_cost;
     int joint;
+    int noadjust;
 
     float alpha;
     float beta;
     float kappa;
 
+    float coord_scale;
+    float object_scale;
+    float noobject_scale;
+    float class_scale;
+
     int dontload;
 
     float probability;
     float scale;
+
     int *indexes;
     float *rand;
     float *cost;
@@ -99,4 +111,6 @@
     #endif
 } layer;
 
+void free_layer(layer);
+
 #endif

--
Gitblit v1.10.0