Spaces:
Build error
Build error
Update lib/dataset/DemoDataset.py
Browse files
lib/dataset/DemoDataset.py
CHANGED
@@ -125,8 +125,9 @@ class LoadImages: # for inference
|
|
125 |
|
126 |
|
127 |
# Padded resize
|
128 |
-
|
129 |
img0 = cv2.resize(img0,(1280,720),interpolation = cv2.INTER_LINEAR)
|
|
|
130 |
img, ratio, pad = letterbox_for_img(img0, new_shape=self.img_size, auto=True)
|
131 |
h, w = img.shape[:2]
|
132 |
shapes = (h0, w0), ((h / h0, w / w0), pad)
|
|
|
125 |
|
126 |
|
127 |
# Padded resize
|
128 |
+
|
129 |
img0 = cv2.resize(img0,(1280,720),interpolation = cv2.INTER_LINEAR)
|
130 |
+
h0, w0 = img0.shape[:2]
|
131 |
img, ratio, pad = letterbox_for_img(img0, new_shape=self.img_size, auto=True)
|
132 |
h, w = img.shape[:2]
|
133 |
shapes = (h0, w0), ((h / h0, w / w0), pad)
|