paddle尝试PP-Vehicle-属性识别

本文介绍了在Windows11与conda环境中,如何使用PP-Vehicle和PP-Human进行车辆和行人的属性识别。涉及安装PaddleDetection、下载模型、配置文件修改及运行代码进行视频和图片测试。在处理过程中,遇到了缺少库的问题,通过安装requirements.txt文件解决。此外,还提供了行人属性的中文标签转化功能。

PP-Vehicle-属性识别

windows11环境,conda虚拟环境中运行。
官网地址:PP-Vehicle

  1. 首先安装环境->

    • conda create -n paddle python=3.7
    • conda activate paddle
      两种方式,一种是缺啥装即:
    • pip install paddlepaddle-gpu
    • pip install pyyaml
    • pip install opencv-python
    • pip install scipy
    • pip install matplotlib
    • pip install scikit-learn
    • pip install pandas
    • pip install tqdm
    • pip install shapely
    • pip install requests
    • pip install lab

    另外一种是:pip install -r requirements.txt

  2. 下载PaddleDetection代码,网址为PaddleDetection

  3. 进入到文件夹内,准备测试的视频cd C:\Users\10106\Desktop\PaddleDetection-release-2.6,随便下载一个汽车的视频,放到了demo/test.mp4

  4. 下载模型,放到pretrained/vehicle_attribute_model下,predtrain是自己创建的。其他模型最好也都下载好,解压一下

  5. 编辑配置文件deploy/pipeline/config/infer_cfg_ppvehicle.yml中VEHICLE_ATTR是用于属性识别的,将其中的model_dir替换为下载的模型。

  6. 修改配置文件中deploy/pipeline/config/infer_cfg_ppvehicle.yml 的VEHICLE_ATTR项的enable: True,以启用该功能。

  7. 运行代码python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_ppvehicle.yml --video_file=demo/test.mp4 --device=gpu,出现了个以下的问题。

  8. 测试单张图片python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_ppvehicle.yml --device=gpu --image_file=demo/car.jpg

  9. 测试文件夹中的图片python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_ppvehicle.yml --device=gpu --image_dir=demo

问题解决

  1. 第一次检测视频运行卡在了这里,单张图片可以。将paddle换成gpu版本后,提示Unable to use JDE/FairMOT/ByteTrace提示要安装lab。主要原因为环境安装时出现问题。最好使用pip install -r requirements.txt进行安装,要不然会缺包。
    在这里插入图片描述

PP-Hum

PP-Vehicle 是一个基于深度学习的车辆分析工具,广泛应用于车辆检测、属性识别、跟踪等任务。它支持多摄像头输入,并能够实现高精度的车辆检测与属性分析,例如车辆类型、颜色、品牌等的识别。该工具基于 PaddlePaddle 框架开发,具有良好的可扩展性和易用性,适合在工业级应用中部署[^2]。 ### 1. 开源与环境准备 PP-Vehicle 是开源项目,可在 GitHub 上找到其源代码。为了运行 PP-Vehicle,首先需要安装 PaddleDetection 和相关依赖项。以下是基本的环境配置步骤: - 安装 PaddlePaddle(建议使用 GPU 版本以提升性能): ```bash pip install paddlepaddle-gpu ``` - 克隆 PaddleDetection 仓库并进入目录: ```bash git clone https://github.com/PaddlePaddle/PaddleDetection.git cd PaddleDetection ``` - 安装依赖包: ```bash pip install -r requirements.txt ``` ### 2. 下载 PP-Vehicle 模型 PP-Vehicle 提供了多种预训练模型,适用于不同的应用场景。可以访问 PaddleDetection 的模型库下载对应模型权重文件。例如,使用 `ppyolo_r50vd_dcn_voc` 模型进行车辆检测: ```bash wget https://paddle-model-ecology.bj.bcebos.com/model/ppvehicle/ppyolo_r50vd_dcn_voc.tar tar -xvf ppyolo_r50vd_dcn_voc.tar ``` ### 3. 运行 PP-Vehicle 进行推理 PP-Vehicle 支持命令行方式和 API 方式进行推理。以下是一个使用命令行方式进行车辆检测的示例: ```bash python ppvehicle.py --model_dir=ppyolo_r50vd_dcn_voc --image_file=path_to_your_image.jpg --device=gpu ``` 其中 `--model_dir` 指定模型路径,`--image_file` 指定输入图像路径,`--device` 可设置为 `gpu` 或 `cpu`。 ### 4. 输出结果与可视化 推理完成后,PP-Vehicle 会输出车辆检测结果,包括车辆的边界框、类别、置信度等信息。同时,也可以通过设置参数 `--draw_threshold` 来控制可视化结果的绘制阈值。 ### 5. 多摄像头与视频流处理 PP-Vehicle 支持多摄像头输入,适用于监控系统等场景。对于视频流处理,可以使用以下命令: ```bash python ppvehicle.py --model_dir=ppyolo_r50vd_dcn_voc --video_file=path_to_your_video.mp4 --device=gpu ``` 其中 `--video_file` 指定视频路径,也可以设置为摄像头设备(如 `--camera_id=0`)以实时处理摄像头输入。 ### 6. 自定义训练 如果需要针对特定场景进行优化,PP-Vehicle 支持自定义训练。用户可以准备自己的数据集,并按照 PaddleDetection 的数据格式进行标注。训练命令如下: ```bash python tools/train.py -c configs/ppyolo/ppyolo_r50vd_dcn_voc.yml ``` 在训练过程中,可以通过 TensorBoard 查看训练日志和损失曲线,以监控训练效果。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值