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

xiaoxiao2021-02-28  54

一. 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 ~/SSDunzip caffe-ssd.zip2.cd ~/SSD/caffe-ssdcp Makefile.config.example Makefile.config针对需求修改几项即可,如下:a.若使用cudnn,则将:#USE_CUDNN := 1 修改成:USE_CUDNN := 1b.若要使用python来编写layer,则将:#WITH_PYTHON_LAYER := 1去掉#号修改为:WITH_PYTHON_LAYER := 1c.修改成:USE_OPENCV := 1, USE_LEVELDB := 1, USE_LMDB := 13.make -j16报错:/usr/include/boost/property_tree/detail/json_parser_read.hpp:257:264: error: ‘type name’ declared as function returning an arraymake: *** [.build_release/cuda/src/caffe/layers/detection_output_layer.o] Error 1解决方案参考以下博客:https://blog.csdn.net/wei_guo_xd/article/details/737294724.make clean make -j16报错:/usr/bin/ld: cannot find lopenblas解决方案:参考https://blog.csdn.net/u014696921/article/details/56678520apt-get install liblapack-dev liblapack3 libopenblas-base libopenblas-dev5. make cleanmake -j16编译成功make pycaffe(注意这里编译前并没有下载相应的库,因为之前已经编译过pycaffe了,编译后也并没有设置环境变量)make test -j16make runtest -j166.解压数据集将下载好的预训练模型VGG_ILSVRC_16_layers_fc_reduced.caffemodel放在/home/iscas/SSD/caffe-ssd/models/VGGNet文件夹下将下载好的数据放在/home/iscas/SSD/caffe-ssd/data文件夹下目前在/home/iscas/SSD/caffe-ssd文件夹下,  cd data  tar -xvf VOCtrainval_11-May-2012.tar  tar -xvf VOCtrainval_06-Nov-2007.tar  tar -xvf VOCtest_06-Nov-2007.tar7. 到caffe-ssd目录下: cd ..更改/home/iscas/SSD/caffe-ssd/data/VOC0712文件夹下create_list.sh中的root_dir=$HOME/SSD/caffe-ssd/data/VOCdevkit/更改/home/iscas/SSD/caffe-ssd/data/VOC0712文件夹下create_dat.sh中的data_root_dir="$HOME/SSD/caffe-ssd/data/VOCdevkit"然后执行:./data/VOC0712/create_list.sh执行: ./data/VOC0712/create_data.sh报错:AtributeEerror:'module' object has no attribute 'LabelMap'解决方案:参考https://blog.csdn.net/wei_guo_xd/article/details/73729472或者参考:https://github.com/weiliu89/caffe/issues/4重新配置了python的环境:    a. sudo gedit ~/.bashrc     b. export PYTHONPATH=/home/iscas/SSD/caffe-ssd/python    c. source ~/.bashrc 8.训练python examples/ssd/ssd_pascal.py报错:配置SSD-caffe测试时出现“Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal”解决 解决方案参考:https://blog.csdn.net/xunan003/article/details/78433383

转载请注明原文地址: https://www.6miu.com/read-1464289.html

最新回复(0)