Skip to content

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

Open
@Lars-Codes

Description

Search before asking

  • 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?

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

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions