Skip to content

Commit

Permalink
Cleanup yolo and remove stateless classes (tinygrad#604)
Browse files Browse the repository at this point in the history
* Add AvgPool2d as a layer

* Clean up a bit

* Remove stateless layers in yolo_nn

* More cleanup

* Save label for test

* Add test for YOLO

* Test without cv2

* Don't fail if cv2 not installed

* Better import

* Fix image read

* Use opencv :)

* Don't download the file

* Fix errors

* Use same version

* Set higher confidence

* Why is the confidence so low?

* Start over

* Remove stateless layers

* Remove extra lines

* Revert changes

* Save a few more lines
  • Loading branch information
jla524 authored Feb 27, 2023
1 parent 71ae6e5 commit 0f58c4c
Show file tree
Hide file tree
Showing 4 changed files with 103 additions and 327 deletions.
58 changes: 0 additions & 58 deletions examples/yolo/yolo_nn.py

This file was deleted.

Loading

0 comments on commit 0f58c4c

Please sign in to comment.