From a47874a7e8ee2f0b22b59132e8e807468233dff0 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 14 Jul 2015 06:25:08 +0000
Subject: [PATCH] nochange option in crop
---
src/crop_layer.c | 4 ++++
Makefile | 4 ++--
src/parser.c | 3 +++
src/crop_layer_kernels.cu | 4 ++++
src/layer.h | 1 +
5 files changed, 14 insertions(+), 2 deletions(-)
diff --git a/Makefile b/Makefile
index 32ff4c0..1b42277 100644
--- a/Makefile
+++ b/Makefile
@@ -1,5 +1,5 @@
-GPU=0
-OPENCV=0
+GPU=1
+OPENCV=1
DEBUG=0
ARCH= -arch=sm_52
diff --git a/src/crop_layer.c b/src/crop_layer.c
index 1319021..d9950d6 100644
--- a/src/crop_layer.c
+++ b/src/crop_layer.c
@@ -48,6 +48,10 @@
int dw = rand()%(l.w - l.crop_width + 1);
float scale = 2;
float trans = -1;
+ if(l.noadjust){
+ scale = 1;
+ trans = 0;
+ }
if(!state.train){
flip = 0;
dh = (l.h - l.crop_height)/2;
diff --git a/src/crop_layer_kernels.cu b/src/crop_layer_kernels.cu
index d60cc0a..9df915c 100644
--- a/src/crop_layer_kernels.cu
+++ b/src/crop_layer_kernels.cu
@@ -181,6 +181,10 @@
float scale = 2;
float translate = -1;
+ if(layer.noadjust){
+ scale = 1;
+ translate = 0;
+ }
int size = layer.batch * layer.w * layer.h;
diff --git a/src/layer.h b/src/layer.h
index c319183..29abec0 100644
--- a/src/layer.h
+++ b/src/layer.h
@@ -49,6 +49,7 @@
int objectness;
int does_cost;
int joint;
+ int noadjust;
float alpha;
float beta;
diff --git a/src/parser.c b/src/parser.c
index 893111e..9670715 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -196,7 +196,10 @@
batch=params.batch;
if(!(h && w && c)) error("Layer before crop layer must output image.");
+ int noadjust = option_find_int_quiet(options, "noadjust",0);
+
crop_layer l = make_crop_layer(batch,h,w,c,crop_height,crop_width,flip, angle, saturation, exposure);
+ l.noadjust = noadjust;
return l;
}
--
Gitblit v1.10.0