YOLOv10模型改进-Backbone改进-第53篇: YOLOv10改进策略【Backbone】| VGG16 Backbone替换
一、本文介绍
本文记录的是利用VGG16作为Backbone改进YOLOv10的特征提取部分。VGG16通过堆叠小卷积核实现深度特征提取。
二、VGG16模块介绍
2.1 设计出发点
VGG16通过堆叠多个3x3卷积核,实现更深层次的特征提取。
2.2 模块结构
VGG16块:
- 3x3卷积:特征提取
- 池化:降采样
三、VGG16的实现代码
importtorchimporttorch.nnasnnclassVGG16(nn.Module