- src中新建mylayer.c和mylayer.h,在makefile里OBJ添加mylayer.o
- darknet.h里LAYER_TYPE添加MYLAYER
- parser.c中 [这个函数就用来解析cfg中[convolutional]这种]
- 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);
}
- 1
- 2
- 3
- network.c中
- include “mylayer.h”,
- get_layer_string函数中添加
case MYLAYER:
return "mylayer";
- 1
- 2
- resize_network函数中添加
else if(l.type == MYLAYER)
{
resize_mylayer_layer(arg)
}
- 1
- 2
- 3
- 4