mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

update comments

Browse Source
This commit is contained in:
Glenn Jocher 2021-12-19 18:04:52 +01:00
parent d5fc52132a
commit ecce7c86c7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
classifier.py
View File

@ -8,9 +8,9 @@ Usage-train:
Usage-inference:
from classifier import *
model = torch.load('best.pt', map_location=torch.device('cpu'))['model'].float()
files = Path('../datasets/mnist/test/7').glob('*.png')
for f in list(files)[:10]:
model = torch.load('path/to/best.pt', map_location=torch.device('cpu'))['model'].float()
files = Path('../datasets/mnist/test/7').glob('*.png') # images from dir
for f in list(files)[:10]: # first 10 images
classify(model, size=128, file=f)
"""