spacewalk01/yolov9-face-detection

Have you encountered the following problem?

turingvideosjp opened this issue · 2 comments

turingvideosjp commented

File "/alg_vepfs/private/xxxxx/yolov9/utils/general.py", line 905, in non_max_suppression
device = prediction.device
AttributeError: 'list' object has no attribute 'device'

spacewalk01 commented

Yes. I've fixed it by changing it like following:

yolov9-face-detection/yolov9/utils/general.py

Line 903 in 92956eb

prediction = prediction[0][1] # select only inference output