WongKinYiu / YOLO

An MIT rewrite of YOLOv9
MIT License
292 stars 19 forks source link

🩹 Various fixes to `notebook_inference.ipynb` #30

Closed ferraridamiano closed 3 weeks ago

ferraridamiano commented 3 weeks ago

Description

In this simple commit I just fixed a couple of issues of the notebook. Moreover I changed the default device from cuda to cpu (since this is a simple example I think that all the people, even who does not own a graphics card have to be able to try it out of the box). Moreover I removed a useless empty code section

Type of change

Please delete options that are not relevant.

Checklist:

henrytsui000 commented 3 weeks ago

Hi,

Thanks for reminding me! I also encountered this bug while deploying other models and have fixed it in commit 89c6a27.

Once everything is verified, I will close the PR.

Best regards, Hao Tang Tsui