Joseph Redmon
2015-07-14 a47874a7e8ee2f0b22b59132e8e807468233dff0
nochange option in crop
5 files modified
16 ■■■■ changed files
Makefile 4 ●●●● patch | view | raw | blame | history
src/crop_layer.c 4 ●●●● patch | view | raw | blame | history
src/crop_layer_kernels.cu 4 ●●●● patch | view | raw | blame | history
src/layer.h 1 ●●●● patch | view | raw | blame | history
src/parser.c 3 ●●●●● patch | view | raw | blame | history
Makefile
@@ -1,5 +1,5 @@
GPU=0
OPENCV=0
GPU=1
OPENCV=1
DEBUG=0
ARCH= -arch=sm_52
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;
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;
src/layer.h
@@ -49,6 +49,7 @@
    int objectness;
    int does_cost;
    int joint;
    int noadjust;
    float alpha;
    float beta;
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;
}