强拆YOLO_V3
这里拆解的代码主要来自GitHub eriklindernoren/PyTorch-YOLOv3。
ultralytics版也很流行,但eriklindernoren版更适合初学者。
1.模型可视化
1.1 yolov3各layer可视化
模型通过netron可视化,并稍作整理后显示如下。
别看很复杂,其实主体部分是backbone darknet_53;
darknet可以理解为一堆res_layer,再加上若干下采样块;
其余部分就是直接为yolo_layer服务的,第一个yolo_layer的输入直接为下采样32倍后的feature map,后两个yolo_layer是上采样2倍的结果与前面相同尺寸特征的融合;这有没有点unet的意思?
其中ÿ