基于深度学习的SSD算法目标检测程序的优化与更新
已对Single-Shot MultiBox Detector目标检测模型在Pytorch中的实现进行大幅度优化与更新。新增支持step、cos学习率下降法及adam、sgd优化器选择。学习率根据batch_size自适应调整,同时增加了图片裁剪功能。此外,还新增了mobilenetv2主干的选择、大量注释、可调整参数,对代码的组成模块进行了修改,增加了fps、视频预测及批量预测等功能。欲进行训练,可在百度云下载所需的ssd_weights.pth和主干权值。训练所需的VOC数据集下载地址如下,已包含训练集、测试集及验证集(与测试集一样),无需再次划分。
下载地址
用户评论
这篇文章不仅仅介绍了算法原理,还提到了实际的应用,非常具有实用性。