caffe | 使用caffe SSD制作VOC07112 lmdb数据集

原创于 2024-03-27 05:52:18 发布 · 529 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#caffe #人工智能 #深度学习

分享专栏收录该内容

489 篇文章

订阅专栏

本文详细指导如何从GitHub克隆CaffeSSD项目，配置环境（包括关闭CUDA和CUDNN使用CPU版本），安装并验证Caffe，以及准备VOC2007和2012数据集，将其转换为LMDB格式，以便进行对象检测任务。

部署运行你感兴趣的模型镜像

git clone -b ssd https://github.com/weiliu89/caffe.git caffe_ssd

cd caffe_ssd

cp caffe/Makefile.config caffe_ssd/

# 把 cuda 和 cudnn 关了，用 cpu 版本的就好了
make -j32
make pycaffe

make test -j8
make runtest -j8

vim ~/.bashrc

# 加入
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/path/to/your/caffe_ssd/python

source ~/.bashrc

验证python环境中的 caffe 是不是你的 ssd_caffe

import caffe
print(caffe.__file__)     # 返回 pycaffe 的路径，你就可以知道是哪个caffe了

制作 VOC0712 lmdb

## 1. 下载VOC2007和VOC2012数据集， 放在caffe_ssd/data目录下：

cd data
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar

tar -xvf VOCtrainval_11-May-2012.tar
tar -xvf VOCtrainval_06-Nov-2007.tar
tar -xvf VOCtest_06-Nov-2007.tar

## 2. 创建lmdb格式的数据：

cd caffe_ssd
./data/VOC0712/create_list.sh
./data/VOC0712/create_data.sh

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理