github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

yolov5 v4

Browse Source
This commit is contained in:
George Hotz
2021-01-05 07:56:17 -08:00
parent 62e3a8558c
commit ce77dda805
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/yolov5.py
View File

@@ -4,8 +4,7 @@ import pickle
from extra.utils import fetch, my_unpickle
if __name__ == "__main__":
dat = fetch('https://github.com/ultralytics/yolov5/releases/download/v3.0/yolov5s.pt')
#dat = fetch('https://github.com/ultralytics/yolov5/releases/download/v3.0/yolov5m.pt')
dat = fetch('https://github.com/ultralytics/yolov5/releases/download/v4.0/yolov5s.pt')
import zipfile
fp = zipfile.ZipFile(io.BytesIO(dat))