- 博客(1)
- 收藏
- 关注
基于深度学习的SSD算法目标检测程序
Single-Shot MultiBox Detector目标检测模型在Pytorch当中的实现
进行了大幅度的更新,支持step、cos学习率下降法、支持adam、sgd优化器选择、支持学习率根据batch_size自适应调整、新增图片裁剪。
进行了大幅度的更新,增加了mobilenetv2主干的选择、增加大量注释、增加了大量可调整参数、对代码的组成模块进行修改、增加fps、视频预测、批量预测等功能。
训练所需的ssd_weights.pth和主干的权值可以在百度云下载。
链接: https://pan.baidu.com/s/1iUVE50oLkzqhtZbUL9el9w
提取码: jgn8
VOC数据集下载地址如下,里面已经包括了训练集、测试集、验证集(与测试集一样),无需再次划分:
链接: https://pan.baidu.com/s/1-1Ej6dayrx3g0iAA88uY5A
提取码: ph32
2023-05-06
基于yolov5的深度学习目标检测程序,配置完环境可直接训练模型
基于yolov5算法的深度学习目标检测程序。YOLOV5:You Only Look Once目标检测模型在pytorch当中的实现(edition v5.0 in Ultralytics)支持多GPU训练,新增各个种类目标数量计算,新增heatmap;将正样本匹配过程加入dataloader,加快了运算速度;加入EMA效果变好。
2022-02:仓库创建,支持不同尺寸模型训练,分别为s、m、l、x版本的yolov5、支持step、cos学习率下降法、支持adam、sgd优化器选择、支持学习率根据batch_size自适应调整、新增图片裁剪。
2023-05-06
基于深度学习的yolov4目标检测程序
基于深度学习的yolov4目标检测程序,配置好环境后可直接使用。
支持多GPU训练,新增各个种类目标数量计算,新增heatmap;将正样本匹配过程加入dataloader,加快了运算速度;加入EMA效果变好。
仓库创建,支持不同尺寸模型训练,分别为s、m、l、x版本的yolov5、支持step、cos学习率下降法、支持adam、sgd优化器选择、支持学习率根据batch_size自适应调整、新增图片裁剪。
2023-05-06
遥感图像数据集(深度学习的遥感图像检测),包含图片和xml标签
一个用于遥感图像中目标检测的开放数据集。数据集包括:飞机数据集,446幅图像中的4993架飞机。游乐场,189张图片中的191个游乐场。天桥,176幅图片中的180座天桥。油箱,165张图片中的1586个油箱。数据集中图片和标签都一一对应且在不同的文件夹。
2023-04-28
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人