
基于深度学习的SSD算法进行目标检测的程序
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
已对SingleShot-MultiBox-Detector目标检测模型在PyTorch平台上的实现进行了较大规模的优化。该模型采用了step和cos学习率衰减策略,并支持Adam或SGD两种优化器的选择;此外,系统还具备根据当前批处理大小自适应调整学习率的功能。同时增加了图像预处理功能,并支持采用MobileNetV2作为主干结构。为了提升代码可读性,新增了详细的注释和多个调节模型超参数的选项。在算法性能方面,对代码架构进行了系统性优化,使得整体运行效率得到显著提升;此外还增加了实时帧率检测、视频预测分析以及批处理预测功能。训练所需的ssd_weights.pth和主干的权值可在百度云下载:https:pan.baidu.coms1iUVE50oLkzqhtZbUL9el9w 提取码: jgn8。VOC数据集下载地址如下,其中包括了完整的数据集,包括训练集、测试集和验证集(其中测试集与验证集内容一致),无需用户额外划分:https:pan.baidu.coms1-1Ej6dayrx3g0iAA88uY5A 提取码: ph32
全部评论 (0)
还没有任何评论哟~


