PaddleDetection开发套件 | VOC格式YOLO3训练

一、环境搭建

• PaddlePaddle 2.2
• OS 64位操作系统
• Python 3(3.5.1+/3.6/3.7/3.8/3.9)，64位版本
• pip/pip3(9.0.1+)，64位版本
• CUDA >= 10.1
• cuDNN >= 7.6

# CUDA10.1
python -m pip install paddlepaddle-gpu==2.2.0.post101 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html

# CPU
python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple

# 在您的Python解释器中确认PaddlePaddle安装成功
>>> import paddle
>>> paddle.utils.run_check()

# 确认PaddlePaddle版本
python -c "import paddle; print(paddle.__version__)"

# 克隆PaddleDetection仓库
cd <path/to/clone/PaddleDetection>
git clone https://github.com/PaddlePaddle/PaddleDetection.git

# 安装其他依赖
cd PaddleDetection
pip install -r requirements.txt

# 编译安装paddledet
python setup.py install

安装后确认测试通过：

python ppdet/modeling/tests/test_architectures.py

测试通过后会提示如下信息：

.......
----------------------------------------------------------------------
Ran 7 tests in 12.816s
OK

有一说一，paddle的安装是真的方便，paddledetection也很简单，比mmlab不知道轻松到哪里去了。

二、准备数据

因为我一直沿用VOC格式数据，所以稍加改动就可以做成paddle需要的样子

如果你并未组织VOC格式的数据集，那么我建议你在标注完成后使用：

YOLO5 训练自定义数据集及避坑指南_訢詡的博客-CSDN博客一、项目地址：https://github.c