https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.7/docs/tutorials/INSTALL_cn.md
conda创建环境
https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html:
根据cuda版本安装paddlepaddle:https://www.paddlepaddle.org.cn/
# 克隆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
快速体验目标检测效果
在GPU上预测一张图片
export CUDA_VISIBLE_DEVICES=0 # windows上不用加
python tools/infer.py -c configs/ppyolo/ppyolo_r50vd_dcn_1x_coco.yml -o use_gpu=true weights=https://paddledet.bj.bcebos.com/models/ppyolo_r50vd_dcn_1x_coco.pdparams --infer_img=demo/000000014439.jpg
更改visualizer.py的代码
计算文本框大小
bbox = draw.textbbox((0, 0), text)
计算文本宽度和高度
tw = bbox[2] - bbox[0]
th = bbox[3] - bbox[1]
结果: