notebooks icon indicating copy to clipboard operation
notebooks copied to clipboard

Need an extra underscore for extra parameter while looping through "detections"

Open Lars-Codes opened this issue 9 months ago • 0 comments

Search before asking

  • [X] I have searched the Roboflow Notebooks issues and found no similar bug report.

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?

  • [X] Yes I'd like to help by submitting a PR!

Lars-Codes avatar May 30 '24 20:05 Lars-Codes