ssd训练自己的数据集

AI摘要

本文档详细介绍了如何使用SSD算法在PyTorch中训练自己的数据集,包括数据集准备、源码修改、训练过程以及在训练过程中可能遇到的问题及其解决方案,如数据集格式、类别数设置、报错处理等。

基于SSD算法实现对自己数据集的训练与检测。(该专题以操作为主)

SSD是一种非常优秀的one-stage目标检测方法,one-stage算法就是目标检测和分类是同时完成的,其主要思路是利用CNN提取特征后,均匀地在图片的不同位置进行密集抽样,抽样时可以采用不同尺度和长宽比,物体分类与预测框的回归同时进行,整个过程只需要一步,所以其优势是速度快。

这篇文档主要讲述怎样用SSD算法来实现对自己数据集的训练步骤,以及一些可能遇到问题的解决方案。

源码采用开源代码,链接如下:

https://github.com/bubbliiiing/ssd-pytorch

代码结构如下图所示

1、准备数据集

需要准备VOC格式的数据集(以VOC格式数据集为例)

 2、修改相应的数据处理函数

① 本文采用的是 自己的数据集,

对比实验系列:SSD环境配置及训练自己数据集 可以从下方的链接下载SSD-Pytorch源码。 阅读详情

相关推荐

SSD-Pytorch训练自己的数据集

在115行,N = num_pos.data.sum()的后面加上.double,继续在下面添加。在204行, 修改第n次迭代保存一次权重文件,将5000改为500,更改权重名称(随意)num_classes类的个数改为自己设定的+1(自己的类别数+背景类别)将train文件中所有的nn.init.xavier_uniform。在28行处,修改VOC_ROOT,为存放自己数据集的路径。更改为nn.init.xavier_uniform_在20行处,修改VOC_CLASSES为自己的类别。

weixin_67856277的博客 2704

SSD安装及训练自己的数据集

最近一直在搞object detection玩,之前用的是faster-rcnn,准确率方面73.2%,效果还不错,但是识别速度有点欠缺,我用的GPU是GTX980ti, 识别速度大概是15fps.最近发现SSD(single shot multibox detector) 这篇论文效果和速度都不错,我自己实验了一下,速度确实比faster-rcnn快不少。下面分两部分来介绍。第一部分介绍SSD的安

zhang_shuai12的专栏 4万+

Tensorflow下使用SSD训练自己的数据集

Tensorflow下使用SSD训练自己的数据集 1、数据集格式转换。 ① 将自己的数据集做成VOC2007格式,直接将VOC2007文件夹粘贴到SSD-Tensorflow-master目录下。 ② 修改datasets文件夹中pascalvoc_common.py文件中的训练类。 #原始的 # VOC_LABELS = { # 'none': (0, 'Background'...

clover_my的博客 1万+

MobileNet SSD 训练自己的数据集

记录下mobilnet-ssd如何跑自己的数据集。默认环境已经配置好,并且demo.py已经可以成功运行了。 mobilenet-ssd下载路径(需要注意的点:需要以Git的方式下载,不要以zip的方式下载,否则后面训练时可能报错) $ git clone https://github.com/chuanqi305/MobileNet-SSD.git 步骤记录:1、制作VOC数据集    ...

小猫猫的博客 1万+

SSD-Pytorch训练和测试自己的数据集(新手必看)

目录 1.准备自己的数据集 2.pytorch环境安装即SSD-pytorch代码下载 3.正常的修改 修改1:config.py 修改2:VOC0712.py 修改3:train.py 修改4:eval.py 修改5:ssd.py 4.遇到的问题和相应的解决办法 问题1: ...

kaimingD的博客 4万+

成功运行SSD_Pytorch并训练自己的数据集

目录前言一、环境二、准备数据集1.voc2007数据集2.voc2012数据集三、从Github上下载作者源码创建文件夹四、修改代码1.修改data/config.py2.修改data/voc0712.py3.修改layers/modules/multibox_loss.py4.修改train.py5.修改ssd.py五、开始训练六、运行eval.py1.修改eval.py2.修改ssd.py七、运行test.py修改test.py八、运行demo.ipynb修改demo.py总结 前言 本文主要是记录自己

m0_47452894的博客 1万+

SSD的运行及训练自己数据集

ubuntu18.04下SSD的运行及训练自己数据集caffe的SSD运行及训练自己数据集运行一、下载源码二、文件配置三、编译 caffe的SSD运行及训练自己数据集 运行 一、下载源码 git clone https://github.com/weiliu89/caffe.git 下载完成后是一个caffe文件夹,如图: 然后执行: cd caffe git checkout ssd 出现...

medusa_zj的博客 2389

MobileNet-SSD训练自己的数据集

一、配置 ubuntu18.04 ,显卡驱动,cuda10.1,cudnn7.6.5 查看cuda版本: cat /usr/local/cuda/version.txt 查看cudnn版本: cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 二、安装配置Caffe-SSD-GPU annaconda下创建caffe环境 conda create -n caffe2 python=2.7 opencv安装:pip install ope

Sunhansong的博客 2061

SSD-tensorflow512训练自己的数据集

1.下载ssd-tensorflow工程压缩包 Github:https://github.com/balancap/SSD-Tensorflow 2.解压 checkpoints文件夹下ssd_300_vgg.ckpt压缩文件删除,将下载的ssd512模型放在此处。 链接:https://pan.baidu.com/s/13rXshm9s8D-7BjX0bNhpkg提取码:f0k5 ...

improving 2965

用SSD训练自己的数据集(VOC2007格式)

用SSD训练自己的数据集(VOC2007格式)一. 配置caffe环境ubunt16.04下caffe环境安装二. 下载,编译及测试ssd源码(一)下载源码github链接或者执行 git clone https://github.com/weiliu89/caffe.git(二)编译源码编译过程中可能会出现缺失一些python模块,按提示安装(1)修改Markfile配置参考ubunt16.04下

Howie的博客 1万+

SSD训练自己的数据集pytorch实现!!踩坑实录

SSD训练自己的数据集keras实现!!踩坑实录

weixin_43935669的博客 1万+

历年北京理工大学复试专业综合题

。

A place to record sth. 2万+

射频接收机本振相噪指标计算:从倒易混频到GSM实战

在射频接收机设计中,相位噪声是衡量本振信号频谱纯度、影响系统灵敏度的关键指标。其核心影响机制源于倒易混频原理:当不纯净的本振信号与强带外阻塞信号混频时,本振的噪声边带会被转换到中频通带内,从而抬升接收机噪声基底。这一过程直接决定了接收机在阻塞测试等场景下的性能极限。从技术价值看,精确计算本振相噪要求,是进行系统噪声预算、实现成本与性能权衡的核心依据。尤其在手机、基站等对灵敏度要求极高的应用中,它指导着锁相环和压控振荡器的选型与设计。本文以经典的GSM DCS1800标准为例,通过详细的公式推导与变量分析,演

weixin_30368405的博客 324

SAP S/4HANA BP主数据-CVI模型

SAP BP主数据介绍 ECC系统中的客户、供应商主数据的限制: •一个客户或供应商仅能有一个地址 •客户和供应商没有关联关系(同一个实体公司) •如果一家公司既是供应商也是客户需建立两个主数据(数据冗余) •属性无时间相关性 S4系统中BP优点: •一个BP可以扮演多个角色(个人,客户,供应商)且角色之间数据共享 •时间相关的属性 •统一的数据架构,无数据冗余 •CVI模型可以确保BP和客户/供应商数据同步 SAP BP主数据配置 •配置BP角色/角色类别 •配置BP角.

qq_37785505的博客 1549

Metal ECO实战:如何在不修改底层掩模的情况下修复芯片时序问题

本文深入探讨了Metal ECO技术在芯片设计中的应用,这是一种在不修改底层晶体管掩模的前提下,仅通过调整金属连线和替换预置填充单元来修复时序问题的“微创手术”。文章详细解析了其原理、适用场景、关键操作流程与命令,并分享了实战中的高级技巧与常见陷阱,为面临流片前时序违例的工程师提供了宝贵的修复指南。

wdx0123456的博客 430

YOLOv8损失函数升级:从CIoU到Wise-IoU v3的实战调优与避坑指南

本文详细介绍了如何将YOLOv8的损失函数从CIoU升级到Wise-IoU v3,以提升目标检测性能,特别是在小目标检测和样本不均衡场景下的表现。通过实战调优步骤和避坑指南,帮助开发者有效应用Wise-IoU v3的动态聚焦机制和单调性控制,显著提升模型精度和训练稳定性。

weixin_33737774的博客 590

文献阅读:SAR像素偏移的性能改进An Improvement of the Performance of SAR Offset Tracking Approach

像素偏移追踪POT文献阅读

weixin_56301271的博客 341

Matlab实战:水声信号DEMON谱分析从入门到精通(附完整源码)

本文详细介绍了使用Matlab进行水声信号DEMON谱分析的完整流程,从基础理论到实战应用,涵盖信号预处理、包络提取、频谱分析等关键步骤,并附完整源码。特别针对水声通信中的低信噪比环境,提供了多种优化方案和性能提升技巧,帮助读者快速掌握这一水下目标检测的核心技术。

gamma的博客 294

科研绘图系列:R语言circos图(circos plot)

科研绘图系列:R语言circos图(circos plot)

专注生信领域 3430

Devexpress16.2.6破解工具

Devexpress16.2.6破解工具Devexpress16.2.6破解工具Devexpress16.2.6破解工具Devexpress16.2.6破解工具

visio安装包

Microsoft Office Visio 是一款专业的制图软件。利用经过更新的形状、协作工具和数据链接图表,创建专业级图表以简化复杂的信息。

Statistics and Machine Learning Toolbox

Statistics and Machine Learning Toolbox(带书签),MATLAB统计学习工具

老版QQ原生表情gif动态

这是老版的QQ表情,不过全是动态的的,Windows照片查看器是不会动的,可以右击用浏览器打开查看,再说一遍是老版的,只是分享给宁愿用老版也要动态的人。

上一篇: scrapy爬虫并保存数据库
下一篇: faster-rcnn 训练自己的数据集
reset2021
博客等级 码龄5年 561粉丝 415原创
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付:10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

reset2021

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值