Alexey
2018-03-26 11e6f4b081588a08687952bd5ddaaa04fece6144
src/parser.c
@@ -24,6 +24,7 @@
#include "parser.h"
#include "region_layer.h"
#include "reorg_layer.h"
#include "reorg_old_layer.h"
#include "rnn_layer.h"
#include "route_layer.h"
#include "shortcut_layer.h"
@@ -60,6 +61,7 @@
    if (strcmp(type, "[max]")==0
            || strcmp(type, "[maxpool]")==0) return MAXPOOL;
    if (strcmp(type, "[reorg]")==0) return REORG;
   if (strcmp(type, "[reorg_old]") == 0) return REORG_OLD;
    if (strcmp(type, "[avg]")==0
            || strcmp(type, "[avgpool]")==0) return AVGPOOL;
    if (strcmp(type, "[dropout]")==0) return DROPOUT;
@@ -248,6 +250,7 @@
   l.small_object = option_find_int_quiet(options, "small_object", 0);
    l.softmax = option_find_int(options, "softmax", 0);
   l.focal_loss = option_find_int_quiet(options, "focal_loss", 0);
    //l.max_boxes = option_find_int_quiet(options, "max",30);
    l.jitter = option_find_float(options, "jitter", .2);
    l.rescore = option_find_int_quiet(options, "rescore",0);
@@ -358,6 +361,23 @@
    return layer;
}
layer parse_reorg_old(list *options, size_params params)
{
   printf("\n reorg_old \n");
   int stride = option_find_int(options, "stride", 1);
   int reverse = option_find_int_quiet(options, "reverse", 0);
   int batch, h, w, c;
   h = params.h;
   w = params.w;
   c = params.c;
   batch = params.batch;
   if (!(h && w && c)) error("Layer before reorg layer must output image.");
   layer layer = make_reorg_old_layer(batch, w, h, c, stride, reverse);
   return layer;
}
maxpool_layer parse_maxpool(list *options, size_params params)
{
    int stride = option_find_int(options, "stride",1);
@@ -657,7 +677,9 @@
        }else if(lt == MAXPOOL){
            l = parse_maxpool(options, params);
        }else if(lt == REORG){
            l = parse_reorg(options, params);
            l = parse_reorg(options, params);      }
      else if (lt == REORG_OLD) {
         l = parse_reorg_old(options, params);
        }else if(lt == AVGPOOL){
            l = parse_avgpool(options, params);
        }else if(lt == ROUTE){