Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-03-20 20:31:09 +02:00
parent e7075f2b23
commit 9885903baf
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
utils/datasets.py
View File

@ -73,7 +73,7 @@ class LoadWebcam: # for inference
ret_val, img0 = self.cam.read() ret_val, img0 = self.cam.read()
assert ret_val, 'Webcam Error' assert ret_val, 'Webcam Error'
img_path = 'webcam_%g.jpg' % self.count img_path = 'webcam_%g.jpg' % self.count
img0 = cv2.flip(img0, 1) img0 = cv2.flip(img0, 1) # flip left-right
# Padded resize # Padded resize
img, _, _, _ = letterbox(img0, height=self.height) img, _, _, _ = letterbox(img0, height=self.height)
@ -155,7 +155,10 @@ class LoadImagesAndLabels: # for training
# Load labels # Load labels
if os.path.isfile(label_path): if os.path.isfile(label_path):
labels0 = np.loadtxt(label_path, dtype=np.float32).reshape(-1, 5) # labels0 = np.loadtxt(label_path, dtype=np.float32).reshape(-1, 5) # SLOWER
with open(label_path, 'r') as file:
lines = file.read().splitlines()
labels0 = np.array([x.split() for x in lines], dtype=np.float32)
# Normalized xywh to pixel xyxy format # Normalized xywh to pixel xyxy format
labels = labels0.copy() labels = labels0.copy()