使用PaddleDetection_pphuman进行行人属性识别如何增减属性?

paddlepaddle版本:2.4
paddledetection版本:2.4

工程文件链接_OneDrive

使用方法:搭建好cuda、cudnn、paddlapaddle、pycocotools、paddledetection配置环境之后,解压工程文件,用cmd激活paddlepaddle环境,cd到工程文件目录后输入python deploy/pphuman/pipeline.py --config deploy/pphuman/config/infer_cfg.yml --video_file=test_video.mp4 --device=gpu --enable_attr=True就可以进行行人属性识别,输出文件在output文件夹。
这是目前的效果图,会在每个行人检测框里显示各自各自穿着、性别等。

img

问题和期望

但是现在只需要识别脸部的属性,只留下眼镜、帽子之类的属性,其他衣服、鞋子之类的属性删掉,再加一个口罩的属性就行。网上找了几篇文章也没找到对的方法,我自己根据命令行找到了output_inference\strongbaseline_r50_30e_pa100k\infer_cfg.yml,里面有属性列表,修改了再跑程序也没有任何效果。

img

如果有人知道怎么增减属性的希望能教一下,效果好的会另外加钱,辛苦了。

参数里应该有选择哪几类的代码,根据属性列表选择所需要的对应index吧,如[0 1 …]

参考PP-Human之行人属性识别开发文档
output_inference\strongbaseline_r50_30e_pa100k\infer_cfg.yml文件里面有属性列表,删掉之后再把deploy\python\attr_infer.py里面的postprocess函数有关这些属性的代码全部删掉,就可以让识别之后的视频不显示这些属性,但是要增添属性要怎么操作还不清楚,没有找到训练集数据,不知道怎么给训练集图片加上新属性用来识别。