基于改进YOLOv5s的无人机小目标检测算法
宋耀莲,王粲,李大焱,刘欣怡
UAV small target detection algorithm based on improved YOLOv5s
Yaolian SONG,Can WANG,Dayan LI,Xinyi LIU
表 1
模型训练参数设置
Tab.1
Model training parameter setting
参数
数值
参数
数值
批量大小
32
余弦退火参数
0.01
训练轮数
150
学习率动量
0.937
图片尺寸
640×640
权重衰减系数
0.0005
初始学习率
0.01
—
—