notebooks icon indicating copy to clipboard operation
notebooks copied to clipboard

Update train-yolo-nas-on-custom-dataset.ipynb

Open Lars-Codes opened this issue 8 months ago • 1 comments

See documentation for verification https://supervision.roboflow.com/detection/core/

Description

Under "Visualize Interference Result," there needs to be an extra underscore when looping through detections.

Please delete options that are not relevant.

  • [x] This change requires a documentation update

How has this change been tested, please provide a testcase or example of how you tested the change?

I was following the tutorial and was receiving errors until I added an extra underscore. As per the documentation, there are 6 parameters, not 5.

Code

Change this:

labels = [ f"{result.class_names[class_id]} {confidence:0.2f}" for _, _, confidence, class_id, _ in detections ] to This:

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

Lars-Codes avatar Jun 18 '24 13:06 Lars-Codes