将Caltech数据集转化为caffe的输入数据格式LMDB请参考上一篇文章：https://blog.csdn.net/sunshine_zkf/article/details/86173247

前言

修改SSD代码，将其适用于只有一类行人的Caltech数据集上。平台Ubuntu16.04+caffe+SSD+python2.7

一、训练和测试SSD网络所需的文件

在caffe_SSD_root/examples/ssd目录下：

ssd_pascal.py ：用于训练VOC数据集,训练自己的数据集基本都是改这个文件。

通过分析ssd_pascal.py的源码，可以知道训练ssd模型需要几个文件输入，分别是

train_data = "examples/VOC0712/VOC0712_trainval_lmdb"
test_data = "examples/VOC0712/VOC0712_test_lmdb"
name_size_file = "data/VOC0712/test_name_size.txt"
pretrain_model = "models/VGGNet/VGG_ILSVRC_16_layers_fc_reduced.caffemodel"  //是base network部分(VGG_16的卷积层)的预训练参数。
label_map_file = "data/VOC0712/labelmap_voc.prototxt"  //保存的是物体的name和label的映射文件，用于训练和测试下面的四个文件由ssd_pascal.py脚本中根据模型定义和训练策略参数自动生成的
train_net_file = "models/VGGNet/VOC0712/SSD_300x300/train.prototxt"
test_net_file = "models/VGGNet/VOC0712/SSD_300x300/test.prototxt"
deploy_net_file = "models/VGGNet/VOC0712/SSD_300x300/deploy.prototxt"
solver_file = "models/VGGNet/VOC0712/SSD_300x300/solver.prototxt"

二、修改SSD网络的文件用于训练Caltech数据集

1、修改ssd_pascal.py

复制ssd_pascal.py，重命名为ssd_pascal_caltech.py

主要就是路径的修改，修改如下：

（1）修改train_data和test_data路径

（2）主要还是路径的修改，和num_classes数量的修改

（3）修改num_test_image的数量

2、训练SSD网络及其结果

在caffe_SSD_root根目录下运行：

python examples/ssd/ssd_pascal_caltech.py

运行结果不收敛：

解决办法：不收敛的原因主要是学习率太高了，降低学习率为base_lr=0.000004，经过120000次迭代后的结果是：

目标检测算法SSD用于行人检测（二）：训练和测试SSD网络相关推荐

目标反射回波检测算法及其FPGA实现之二：互相关/卷积/FIR电路的实现
目标反射回波检测算法及其FPGA实现之二: 互相关/卷积/FIR电路的实现前段时间,接触了一个声呐目标反射回波检测的项目.声呐接收机要实现的核心功能是在含有大量噪声的反射回波中,识别出发射机发出的激 ...
基于光电检测图像处理目标检测算法相较传统检测技术的优势
光电检测图像处理目标检测算法相较传统检测技术有以下几个优势: 速度更快:光电检测图像处理算法可以在较短的时间内处理大量的图像数据,而传统检测技术需要更长的时间来进行目标检测. 精度更高:光电检测图像处 ...
路面裂痕检测YOLO算法、目标检测算法实现地面裂缝检测
道路裂纹检测YOLO算法,目标检测,目标识别,裂纹检测路面裂痕检测YOLO算法.目标检测算法实现地面裂缝检测车头定位交通标志识别车道线识别自己标注数据,训练模型,效果很好4360063193 ...
论文翻译 SGCN:Sparse Graph Convolution Network for Pedestrian Trajectory Prediction 用于行人轨迹预测的稀疏图卷积网络
SGCN:Sparse Graph Convolution Network for Pedestrian Trajectory Prediction 用于行人轨迹预测的稀疏图卷积网络行人轨迹预测是自 ...
基于CNN目标检测方法（RCNN，Fast-RCNN，Faster-RCNN，Mask-RCNN，YOLO，SSD）行人检测，目标追踪，卷积神经网络
一.研究意义卷积神经网络(CNN)由于其强大的特征提取能力,近年来被广泛用于计算机视觉领域.1998年Yann LeCun等提出的LeNet-5网络结构,该结构使得卷积神经网络可以端到端的训练,并应 ...
扫描式SSD用于公式检测，代码已开源
SSD是大家常见的目标检测器,在类似COCO的日常生活类物体检测,或者人脸.行人检测上都有应用,不过是否可以用于文本领域的检测呢? 今天跟大家介绍一篇新出的论文 ScanSSD: Scanning S ...
ICDAR2013文本检测算法的衡量方法（二）Rectangle Matching与DetEval
这是第二部分,第一部分请点击 ICDAR2013文本检测算法的衡量方法(一)Evaluation Levels 主要内容: Rectangle Matching, 或者bbox matching IC ...
自动泊车之停车位检测算法（角点检测/语义分割）
作者丨机器学习AI算法工程来源丨希骥智能网联汽车点击进入->3D视觉工坊学习交流群一.背景介绍自动泊车大体可分为4个等级: 第1级,APA 自动泊车:驾驶员在车内,随时准备制动,分为雷达 ...
行人检测（上）-行人检测介绍
原文链接:http://www.360doc.com/content/15/1225/09/2459_522949183.shtml 行人检测具有极其广泛的应用:智能辅助驾驶,智能监控,行人分析以及智 ...

目标检测算法SSD用于行人检测（二）：训练和测试SSD网络

前言

一、训练和测试SSD网络所需的文件

二、修改SSD网络的文件用于训练Caltech数据集

1、修改ssd_pascal.py

2、训练SSD网络及其结果

目标检测算法SSD用于行人检测（二）：训练和测试SSD网络相关推荐

最新文章

热门文章