Quellcode durchsuchen

Update Hub Path inputs (#4200)

modifyDataloader
Glenn Jocher GitHub vor 3 Jahren
Ursprung
Commit
2683b18079
Es konnte kein GPG-Schlüssel zu dieser Signatur gefunden werden GPG-Schlüssel-ID: 4AEE18F83AFDEB23
2 geänderte Dateien mit 5 neuen und 3 gelöschten Zeilen
  1. +3
    -1
      hubconf.py
  2. +2
    -2
      models/common.py

+ 3
- 1
hubconf.py Datei anzeigen

@@ -115,9 +115,11 @@ if __name__ == '__main__':
import cv2
import numpy as np
from PIL import Image
from pathlib import Path

imgs = ['data/images/zidane.jpg', # filename
'https://github.com/ultralytics/yolov5/releases/download/v1.0/zidane.jpg', # URI
Path('data/images/zidane.jpg'), # Path
'https://ultralytics.com/images/zidane.jpg', # URI
cv2.imread('data/images/bus.jpg')[:, :, ::-1], # OpenCV
Image.open('data/images/bus.jpg'), # PIL
np.zeros((320, 640, 3))] # numpy

+ 2
- 2
models/common.py Datei anzeigen

@@ -2,7 +2,7 @@

import logging
from copy import copy
from pathlib import Path, PosixPath
from pathlib import Path

import math
import numpy as np
@@ -248,7 +248,7 @@ class AutoShape(nn.Module):
shape0, shape1, files = [], [], [] # image and inference shapes, filenames
for i, im in enumerate(imgs):
f = f'image{i}' # filename
if isinstance(im, (str, PosixPath)): # filename or uri
if isinstance(im, (str, Path)): # filename or uri
im, f = Image.open(requests.get(im, stream=True).raw if str(im).startswith('http') else im), im
im = np.asarray(exif_transpose(im))
elif isinstance(im, Image.Image): # PIL Image

Laden…
Abbrechen
Speichern