PP-Vehicle-属性识别
windows11环境,conda虚拟环境中运行。
- 首先安装环境
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
- 下载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
问题解决
- 第一次检测视频运行卡在了这里,单张图片可以。将paddle换成gpu版本后,提示Unable to use JDE/FairMOT/ByteTrace提示要安装lab,
pip install lab
,but问题并没有解决,output文件夹已经生成了一帧的视频,猜测是读取视频流出了问题。先做记录。