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

5.6k stars 882 forks source link

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

Closed Lars-Codes closed 2 hours ago

Lars-Codes commented 5 months ago

Description

Under Single Image Interference,

This code: list(model.predict(image, conf=0.35))[0] Should become: model.predict(image, conf=0.35)

Please delete options that are not relevant.

[ ] This change requires a documentation update

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

My code on this notebook only worked when I made this change. Otherwise it was throwing errors.

CLAassistant commented 4 months ago

All committers have signed the CLA.

LinasKo commented 2 hours ago

Hi @Lars-Codes 👋

Thank you for your contribution! Unfortunately, I can’t merge this PR due to the "Invalid Head Notebook In The Diff" issue.

I did encounter the same problems while working on #344 and have addressed them there, so I’ll be closing this PR. Thank you again for taking the time to contribute, and I hope to see more of your work in the future!