updates
This commit is contained in:
parent
c56516ec11
commit
1827b79647
|
@ -74,7 +74,7 @@ class LoadImages: # for inference
|
|||
print('image %g/%g %s: ' % (self.count, self.nF, path), end='')
|
||||
|
||||
# Padded resize
|
||||
img, _, _, _ = letterbox(img0, new_shape=self.height)
|
||||
img, *_ = letterbox(img0, new_shape=self.height)
|
||||
|
||||
# Normalize RGB
|
||||
img = img[:, :, ::-1].transpose(2, 0, 1) # BGR to RGB
|
||||
|
@ -116,7 +116,7 @@ class LoadWebcam: # for inference
|
|||
print('webcam %g: ' % self.count, end='')
|
||||
|
||||
# Padded resize
|
||||
img, _, _, _ = letterbox(img0, new_shape=self.height)
|
||||
img, *_ = letterbox(img0, new_shape=self.height)
|
||||
|
||||
# Normalize RGB
|
||||
img = img[:, :, ::-1].transpose(2, 0, 1) # BGR to RGB
|
||||
|
|
Loading…
Reference in New Issue