PP-Vehicle-属性识别
windows11环境,conda虚拟环境中运行。
官网地址:PP-Vehicle
-
首先安装环境->
conda create -n paddle python=3.7conda activate paddle
两种方式,一种是缺啥装即:pip install paddlepaddle-gpupip install pyyamlpip install opencv-pythonpip install scipypip install matplotlibpip install scikit-learnpip install pandaspip install tqdmpip install shapelypip install requestspip install lab
另外一种是:
pip install -r requirements.txt -
下载PaddleDetection代码,网址为PaddleDetection
-
进入到文件夹内,准备测试的视频
cd C:\Users\10106\Desktop\PaddleDetection-release-2.6,随便下载一个汽车的视频,放到了demo/test.mp4 -
下载模型,放到
pretrained/vehicle_attribute_model下,predtrain是自己创建的。其他模型最好也都下载好,解压一下 -
编辑配置文件
deploy/pipeline/config/infer_cfg_ppvehicle.yml中VEHICLE_ATTR是用于属性识别的,将其中的model_dir替换为下载的模型。 -
修改配置文件中
deploy/pipeline/config/infer_cfg_ppvehicle.yml的VEHICLE_ATTR项的enable: True,以启用该功能。 -
运行代码
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_ppvehicle.yml --video_file=demo/test.mp4 --device=gpu,出现了个以下的问题。 -
测试单张图片
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_ppvehicle.yml --device=gpu --image_file=demo/car.jpg -
测试文件夹中的图片
python deploy/pipeline/pipeline.py --config deploy/pipeline/config/infer_cfg_ppvehicle.yml --device=gpu --image_dir=demo
问题解决
- 第一次检测视频运行卡在了这里,单张图片可以。将paddle换成gpu版本后,提示Unable to use JDE/FairMOT/ByteTrace提示要安装lab。主要原因为环境安装时出现问题。最好使用
pip install -r requirements.txt进行安装,要不然会缺包。


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

被折叠的 条评论
为什么被折叠?



