Warn save disabled for torch inference (#3361)

This commit is contained in:
Glenn Jocher
2023-06-24 15:57:17 +02:00
committed by GitHub
parent 2ebd808b69
commit 5381fc8a58
2 changed files with 8 additions and 5 deletions

View File

@ -294,10 +294,11 @@ class LoadPilAndNumpy:
class LoadTensor:
def __init__(self, imgs) -> None:
self.im0 = imgs
self.bs = imgs.shape[0]
def __init__(self, im0) -> None:
self.im0 = im0
self.bs = im0.shape[0]
self.mode = 'image'
self.paths = [getattr(im, 'filename', f'image{i}.jpg') for i, im in enumerate(im0)]
def __iter__(self):
"""Returns an iterator object."""
@ -309,7 +310,7 @@ class LoadTensor:
if self.count == 1:
raise StopIteration
self.count += 1
return None, self.im0, None, '' # self.paths, im, self.im0, None, ''
return self.paths, self.im0, None, ''
def __len__(self):
"""Returns the batch size."""