darknet中添加新层


  • src中新建mylayer.c和mylayer.h,在makefile里OBJ添加mylayer.o
  • darknet.h里LAYER_TYPE添加MYLAYER
  • parser.c中

  • include “mylayer.h”,
  • string_to_layer_type函数中添加 if(strcmp(type, “[mylayer]”) ==0) return MYLAYER
  • 添加parse_mylayer函数
  • parse_network_cfg中加入
else if(lt == MYLAYER){
    l = parse_mylayer(arg);
}


  • network.c中

  • include “mylayer.h”,
  • get_layer_string函数中添加

case MYLAYER: 
    return "mylayer";
  • resize_network函数中添加
else if(l.type == MYROUTE)
{ 
    resize_myroute_layer(arg)
}

你可能感兴趣的:(深度学习)