paddleDetecion部署流程
* OS:Ubuntu20.04 * python 3.9.13 * CUDA version:11.6 * cudnn version:8.6
conda install paddlepaddle-gpu==2.4.1 cudatoolkit=11.6 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/ -c conda-forge
import paddle paddle.utils.run_check()
git clone https://github.com/PaddlePaddle/PaddleDetection.git git clone https://ghproxy.com/https://github.com/PaddlePaddle/PaddleDetection.git ##github无法直接克隆可使用此命令
cd PaddleDetection pip install -r requirements.txt
python setup.py install
python ppdet/modeling/tests/test_architectures.py
Ran 7 tests in 18.430s OK
PaddleDetection中已经集成了ppyolo、yolox等多种目标检测模型,可直接使用。
① 模型导出 (以yolox_tiny为例)
运行以下命令:
python tools/export_model.py -c configs/yolox/yolox_tiny_300e_coco.yml -o weights=https://paddledet.bj.bcebos.com/models/yolox_tiny_300e_coco.pdparams
模型导出至目录output_inference/yolox_tiny_300e_coco/
② 进行目标检测
修改deploy/python/infer.py用以计算fps并运行以下命令
python deploy/python/infer.py --model_dir=output_inference/yolox_tiny_300e_coco/ --video_file=/data/user21262923/bhy/paddle/PaddleDetection/demo/test1.mp4 --device=gpu --thresh=0.2
③ 测试结果
yolox_tiny | 7.24 |
yolox_cdn_tiny | 7 |
yolox_nano | 7 |
ppyoloe_tiny | 19.1 |
>
如果觉得我的文章对您有用,请随意打赏。你的支持将鼓励我继续创作!