目标检测SSD-tensorflow之训练数据集

最新推荐文章于 2021-04-22 10:57:50 发布
最新推荐文章于 2021-04-22 10:57:50 发布
阅读量693 收藏 2
点赞数
CC 4.0 BY-SA版权
文章标签： SSD
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/qq_40806289/article/details/89511095
本文详细介绍如何制作tfrecord数据集，调整SSD模型以适应自定义检测类别，包括修改类别数目、训练步长及加载预训练模型进行再训练的过程。同时，提供了训练结束后模型测试的方法。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

tfrecord数据集的制作     https://blog.youkuaiyun.com/qq_40806289/article/details/89510467

 重新训练模型的代码修改（主要是自己检测类别数目的修改）

1、datasets文件夹下：pascalvoc_2007.py文件

2、nets文件夹下ssd_vgg_300.py，96、97行（类别数+1）

3、根目录下eval_ssd_network.py文件，66行，类别数+1

4、修改训练步长epoch。train_ssd_network.py

27行：

DATA_FORMAT = 'NCHW'  # gpu
# DATA_FORMAT = 'NHWC'    # cpu

135行：类别数+1

154行：讲None（无限训练下去）修改为自己想要的迭代步数<

最低0.47元/天 解锁文章

200万优质内容无限畅学
逼疯代码
关注
SSD-tensorflow512训练自己的数据集
improving
05-28 2897
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 ...
Tensorflow下使用SSD训练自己的数据集
clover_my的博客
03-27 1万+
Tensorflow下使用SSD训练自己的数据集 1、数据集格式转换。 ① 将自己的数据集做成VOC2007格式，直接将VOC2007文件夹粘贴到SSD-Tensorflow-master目录下。 ② 修改datasets文件夹中pascalvoc_common.py文件中的训练类。 #原始的 # VOC_LABELS = { # 'none': (0, 'Background'...
SSD-Tensorflow 训练自己的数据集
Joel丶yang的专栏
05-25 5058
阅读此文默认你已经可以成功运行SSD-TensorFlow的官方源码，也就是能够在VOC2007数据集上成功训练。下面将介绍一下如何用SSD-Tensorflow 训练自己的数据。如果不清楚的话可以参考链接：https://github.com/balancap/SSD-Tensorflow https://blog.youkuaiyun.com/ei1990/article/details/75282855...
使用自己的数据训练MobileNet SSD v2目标检测--TensorFlow object detection
weixin_43056275的博客
04-21 3664
使用自己的数据训练MobileNet SSD v2目标检测--TensorFlow object detection1. 配置1.1 下载models-1.12.02. 准备数据集3. 配置文件和模型3.1 下载预训练模型3.2 修改配置文件4. 训练4.1 使用tensorboard查看训练过程5. 冻结模型参数6. 调用pb文件进行预测 1. 配置 基本配置 版本号 CPU I...
SSD目标检测训练自己数据实现及遇到问题
chunzhen2020的博客
12-22 212
链接 https://blog.youkuaiyun.com/After__today/article/details/108448263 找了很多，此博客里面可以实现，感谢博主。 遇到问题： 1）tensorflow2.0，不包含tf.contrib, 需要自己安装slim https://blog.youkuaiyun.com/qq_39501363/article/details/107238386 2）tensorflow2.0，no attribute ‘placeholder’, 需替换 https://bl
SSD-Tensorflow-master.zip
06-09
用户可以根据自己的数据集调整参数，例如批量大小、学习率、训练迭代次数等。 5. **模型评估与转换**： `eval_ssd_network.py`用于评估训练好的模型的性能，通常会计算平均精度（mAP）等指标。`caffe_to_...
目标检测SSD-tensorflow之数据集tfrecord的制作
qq_40806289的博客
04-25 1421
SSD-tensorflow github： https://github.com/balancap/SSD-Tensorflow 1、数据集介绍（仿照VOC2007数据集制作自己的数据集） 其中Annotations中存放标签文件xml（使用labelImg打标签） JPEGImages中存放原图 。 ImageSets中存放各个图片的索引，即哪些用来训练，哪些用来测...
使用SSD训练自己的模型（从图片标注开始）
dingding的专栏
04-08 3925
此文章参考了https://blog.youkuaiyun.com/zzZ_CMing/article/details/81131101 在此表示感谢，如果有侵权的地方可联系本人删除 训练手表模型步骤 制作数据集 在项目下面新建文件夹VOC2007，在此文件夹下新建3个文件夹，如下图 JPEGImages：用于存放训练、测试的图片（图片格式最好为.jpg） Annotations：用于存放.xm...
VGG_VOC0712_SSD_300x300_iter_120000.caffemodel
08-21
VGG_VOC0712_SSD_300x300_iter_120000.caffemodel，caffe-SSD下已经训练好的VGG模型文件，希望对大家有所帮助。
深度学习之目标检测（六）SSD
DDDDavid的博客
04-22 514
深度学习之目标检测（六）SSD目标检测之SSDSSD流程：参考链接SSD实际操作①下载代码和模型②运行demo③处理数据集问题记录④模型训练⑤训练模型调整 目标检测之SSD SSD流程： 1、输入图片，到与训练好的分类网络，生成不同大小的特征映射； 2、抽取卷积层的feature_map，构造bbox，进行检测和分类； 3、不同的bbox结合，通过NMS删除不正确的bbox，达到最终的bbox。 参考链接 目标检测之SSD 官方代码地址 文章地址 权重文件地址 代码地址 balancap/SSD-Tenso
ssd权重文件
12-05
ssd,预训练文件。
SSD-300 VGG-based weights 权重包含两种ssd300的权重
11-18
SSD_300_vggmodel，包含两种ssd300： Model Training data Testing data mAP SSD-300 VGG-based VOC07+12+COCO trainval VOC07 test SSD-300 VGG-based VOC07+12 trainval VOC07 test -
SSD-300 VGG-based
08-05
Note: SSD300* and SSD512* are the latest models. Current code should reproduce these results.
tensorflow+SSD300+python3
weixin_41463944的博客
08-18 527
一、准备 搭建SSD框架，下载解压即可 下载pascalvoc数据，自己的数据根据voc格式改写（图片的名称，不用拘泥于6位数字，其他命名也可以）数据集下载点击 解压后不要混合在一个文件夹下 VOCtrainval用来训练，VOCtest用来测试。 VOCtrainval 中JPEGImage文件夹中仅是训练和验证的图片，Main文件夹中仅是trainval.txt, train.txt, val...
目标检测SSD+Tensorflow 训练自己的数据集
热门推荐
06-04 6万+
1.代码地址：https://github.com/balancap/SSD-Tensorflow，下载该代码到本地 2.解压ssd_300_vgg.ckpt.zip 到checkpoint文件夹下 3.测试一下看看，在notebooks中创建demo_test.py,其实就是复制ssd_notebook.ipynb中的代码，该py文件是完成对于单张图片的测试，对Jupyter不熟，就自己...
tensorflow SSD训练自己的数据集
蒋程扬的部落格
01-10 1381
本篇文章是我在自己学习的过程中写的，当作一个笔记，写的比较详细 在github上下载SSD的tensorflow版本： https://github.com/balancap/SSD-Tensorflow 下载后先解压缩checkpoint： unzip ssd_300_vgg.ckpt.zip 用jupyter-notebook 跑一下训练好的模型： jupyter notebook note...
SSD-Tensorflow 目标检测（自定义数据集（VOC2007格式））
红叶谷 wsp_1138886114的博客
12-16 6633
一、准备 搭建SSD框架，下载解压即可 下载pascalvoc数据，自己的数据根据voc格式改写（图片的名称，不用拘泥于6位数字，其他命名也可以）数据集下载点击 解压后不要混合在一个文件夹下 VOCtrainval用来训练，VOCtest用来测试。 VOCtrainval 中JPEGImage文件夹中仅是训练和验证的图片，Main文件夹中仅是trainval.txt, train.txt...
制作SSD目标检测模型需要的训练数据并训练SSD目标检测模型
maweifei的博客
07-17 2264
1构建 数据集 先来看一下我们构建数据集合应该是什么样的,假设总数据为1000张。  为了方便，我们将数据放在／home／bingolwang/data 文件夹下。/home/bingolwang/data/VOCdevkit 这个目录下是VOC2007 VOC2007/ |-- Annotations #1000个xml文件。 |-- ImageSets | `-- Main ...
深度学习在医学图像上应用
yunxinan的专栏
07-22 6503
细胞病理学识别和疾病组织目标检测是目标人工智能技术在影像医学和病理方向的重要应用。 该技术主要是前期的预处理技术复杂，主要原因是因为医学的相关病理特征成因复杂，图像方面的随机误差很大（噪音），图像断层之间的重叠。例如如果对图像的颜色没有识别要求，为了训练和计算方便采用的降维处理是将其灰化处理，然后使用分割算法将其不同的形态结构分离，该过程包含先将其腐蚀再膨胀然后过滤（一般情况中值滤波能去...
深度学习框架SSD-Tensorflow模型权重包下载
综上所述，"SSD-Tensorflow.zip"压缩包是一个包含了深度学习目标检测算法SSD的Tensorflow实现，以及在大量数据集上训练得到的预训练权重的压缩文件。这类资源对计算机视觉研究人员和工程师来说极为宝贵，可以快速...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值