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.
[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.
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?