SSD算法理解及利用SSD训练自己的数据

最新推荐文章于 2024-08-12 14:50:42 发布

yanxiaopan

最新推荐文章于 2024-08-12 14:50:42 发布

阅读量2k

点赞数

分类专栏：安装教程深度学习

本文链接：https://blog.csdn.net/yanxiaopan/article/details/79971180

版权

本文详细介绍了SSD算法的理解，并通过步骤演示如何在Ubuntu环境下安装SSD-Caffe，解决编译过程中的问题，配置数据集进行训练。包括：安装依赖、修改Makefile.config、编译Caffe、解压数据、创建数据列表和数据文件、解决运行错误，以及最终的训练过程。

摘要由CSDN通过智能技术生成

一. SSD算法理解

参考博客：

https://blog.csdn.net/u014380165/article/details/72824889

https://blog.csdn.net/wfei101/article/details/78176322

二. 安装并训练自己的数据

1.在https://github.com/weiliu89/caffe/tree/ssd 下载了.zip包和数据集及预训练模型，放在/home/iscas/SSD文件夹下
解压到当前目录：
cd ~/SSD
unzip caffe-ssd.zip
2.
cd ~/SSD/caffe-ssd
cp Makefile.config.example Makefile.config

针对需求修改几项即可，如下：
a.若使用cudnn，则将:#USE_CUDNN := 1 修改成：USE_CUDNN := 1
b.若要使用python来编写layer，则将:#WITH_PYTHON_LAYER := 1去掉#号修改为:WITH_PYTHON_LAYER := 1
c.修改成:USE_OPENCV := 1, USE_LEVELDB := 1, USE_LMDB := 1

3.
make -j16
报错：
/usr/include/boost/property_tree/detail/json_parser_read.hpp:257:264: error: ‘type name’ declared as function returning an array
make: *** [.build_release/cuda/src/caffe/layers/detection_output_layer.o] Error 1
解决方案参考以下