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

PyTorch Hub and autoShape update (#1415)

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