YOLOv5官方自带权重只检测人

在detect.py文件中,

 第228行

'--classes',nargs='+', type=int, help='filter by class: --classes 0, or --classes 0 2 3' 只需要加上default='0' 就可以指定人了

你可能感兴趣的:(python)