YOLOv7算法简化重构
重构项目简介
YOLOv7是很优秀的检测算法,作者将所有构件写入了yaml文件以增强可改造性,但是熟悉了YOLO’v5网络结构的读者,观看YOLOv7还是相对费劲。由于本人最近在改动YOLOv7模型, 改动起来还是不如YOLOv5方便,因此按照本人自身想法,对YOLOv7网络进行重构。
1. 重构前后参数量和计算量均相同 2. 重构前105层,重构后31层,性能完全相同重构前:
重构后:
重构后的核心代码
# 重构YOLOv7-----------------------------------------------------------------------------
<