ValueError in Labels issues?

[Johan Huynh]

has anyone ever come into the issue below when trying to test other SAM models:
Traceback (most recent call last):
File "E:\PythonProjects\Grounded-Segment-Anything\EfficientSAM\grounded_mobile_sam.py", line 145, in
main(args)
File "E:\PythonProjects\Grounded-Segment-Anything\EfficientSAM\grounded_mobile_sam.py", line 83, in main
labels = [
File "E:\PythonProjects\Grounded-Segment-Anything\EfficientSAM\grounded_mobile_sam.py", line 83, in
labels = [
ValueError: too many values to unpack (expected 5)

Let me know how to fix this please. Thank you!

[haviduck]

detections returns 6 values, while the labels list takes into account 5. fixed by changing:

labels = [ f"{CLASSES[class_id]} {confidence:0.2f}" for _, _, confidence, class_id, _ in detections]

to:

labels = [ f"{CLASSES[class_id]} {confidence:0.2f}" for _, _, confidence, class_id, _, _ in detections]

[Johan Huynh]

thanks a lot :) @haviduck