Procházet zdrojové kódy

PyTorch Hub and autoShape update (#1415)

5.0
Glenn Jocher před 3 roky
rodič
revize
311de00855
1 změnil soubory, kde provedl 2 přidání a 1 odebrání
  1. +2
    -1
      hubconf.py

+ 2
- 1
hubconf.py Zobrazit soubor

@@ -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()

Načítá se…
Zrušit
Uložit