1. 首页
  2. 人工智能
  3. 深度学习
  4. 基于深度学习的SSD算法目标检测程序的优化与更新

基于深度学习的SSD算法目标检测程序的优化与更新

上传者: 2023-07-01 17:26:54上传 ZIP文件 5.33MB 热度 11次

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

用户评论