hank1996 commited on
Commit
a9ca1e4
1 Parent(s): e09d442

Update lib/dataset/DemoDataset.py

Browse files
Files changed (1) hide show
  1. lib/dataset/DemoDataset.py +2 -1
lib/dataset/DemoDataset.py CHANGED
@@ -125,8 +125,9 @@ class LoadImages: # for inference
125
 
126
 
127
  # Padded resize
128
- h0, w0 = img0.shape[:2]
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)