yolox增加一个浅层检测分支
前言
该篇文章主要介绍yolox中如何增加一个检测层,之前的yolov3-yolov5增加检测分支都有人再写,我网上搜索没看到有yolox增加检测分支,所以抽点空闲时间给yolox增加检测一个分支。
一、yolox网络介绍
yolox是在yolov3-yolov5网络上改进而来,所以就不详细介绍了大家可以网上随便看看一看。
二、改进步骤
1.主要改动的几个py文件
这里主要在 yolo_pafpn yolo_head yolo_base darknet 这几个py文件中进行改动。
在yolo_head.py中主要改动在下面代码部分:
在yolo_base.py中的改动如下:
在darknet.py改动如下:
在yolo_pafpn.py改动如下:
总结
yolox增加检测层的步骤基本上就这些,按照上面的改一改就可以训练起来了,大家还有什么改动的想法的可以提评论出来,我来改动一下,像增加注意力机制,该损失函数啥的网上就能搜到,这里就不在继续写了。