提交 311de008 authored 作者: Glenn Jocher's avatar Glenn Jocher

PyTorch Hub and autoShape update (#1415)

上级 2aaaec3c
...@@ -8,7 +8,6 @@ Usage: ...@@ -8,7 +8,6 @@ Usage:
from pathlib import Path from pathlib import Path
import torch import torch
from PIL import Image
from models.yolo import Model from models.yolo import Model
from utils.general import set_logging from utils.general import set_logging
...@@ -112,6 +111,8 @@ if __name__ == '__main__': ...@@ -112,6 +111,8 @@ if __name__ == '__main__':
model = model.fuse().autoshape() # for PIL/cv2/np inputs and NMS model = model.fuse().autoshape() # for PIL/cv2/np inputs and NMS
# Verify inference # Verify inference
from PIL import Image
imgs = [Image.open(x) for x in Path('data/images').glob('*.jpg')] imgs = [Image.open(x) for x in Path('data/images').glob('*.jpg')]
results = model(imgs) results = model(imgs)
results.show() results.show()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论