| | |
| | | #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" |
| | |
| | | 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; |
| | |
| | | |
| | | 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); |
| | |
| | | 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); |
| | |
| | | }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){ |