roboflow / notebooks

Examples and tutorials on using SOTA computer vision models and techniques. Learn everything from old-school ResNet, through YOLO and object-detection transformers like DETR, to the latest models like Grounding DINO and SAM.
https://roboflow.com/models
4.76k stars 742 forks source link

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

Open Lars-Codes opened 2 weeks ago

Lars-Codes commented 2 weeks ago

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.

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 ]