This commit is contained in:
Glenn Jocher
2019-04-22 12:51:20 +02:00
parent 37799efa0b
commit 0bac735cc6
2 changed files with 4 additions and 2 deletions
+1
View File
@@ -75,6 +75,7 @@ class LoadImages: # for inference
# Padded resize
img, _, _, _ = letterbox_rect(img0, height=self.height)
print('%gx%g ' % img.shape[:2], end='') # print image size
# Normalize RGB
img = img[:, :, ::-1].transpose(2, 0, 1) # BGR to RGB