yolov5 转 onnx 模型focus哪去了

yolov5的export.py可以直接把pt转成onnx,用netron查看,奇怪,没有focus层!

查看原始的yolov5s.pt,是有focus层的。

然道,yolov5的focus层有开关?于是开启了网络大搜索,focus层哪去了?几乎所有的大神讲解都有focus层,也没听说有什么参数配置呀!

最后,还是在github的源码介绍上看到了答案,原来已经用一个6*6的卷积层替代了focus层。。。

YOLOv5 (6.0/6.1) brief summary · Issue #6998 · ultralytics/yolov5 · GitHub

Some minor changes compared to previous versions:

  1. Replace the Focus structure with 6x6 Conv2d(more efficient, refer #4825)
  2. Replace the SPP structure with SPPF(more than double the speed)

记录一下,感觉都没人提到

你可能感兴趣的:(深度学习,pytorch,人工智能)