Need an extra underscore for extra parameter while looping through "detections" #279
Open
2 tasks done
Labels
bug
Something isn't working
Search before asking
Notebook name
train-yolo-nas-on-custom-dataset.ipynb
Bug
Under "Visualize Interference Result," there is this for loop:
labels = [ f"{result.class_names[class_id]} {confidence:0.2f}" for _, _, confidence, class_id, _ in detections ]
But it needs an extra underscore for it to function properly:
labels = [ f"{result.class_names[class_id]} {confidence:0.2f}" for _, _, confidence, class_id, _, _ in detections ]
Environment
Ubuntu 22.04, Python 3.8.10
Minimal Reproducible Example
No response
Additional
Parameter list for detections class can be found in this documentation.
Are you willing to submit a PR?
The text was updated successfully, but these errors were encountered: