一、环境搭建
- 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.com/ul