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

Update datasets.py

This commit is contained in:
Glenn Jocher 2019-03-22 14:52:58 +02:00 committed by GitHub
parent cd188dbde6
commit 3532ee038f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
utils/datasets.py
View File

@ -137,20 +137,20 @@ class LoadImagesAndLabels(Dataset): # for training/testing
# 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) # SLOWER
with open(label_path, 'r') as file: with open(label_path, 'r') as file:
lines = file.read().splitlines() lines = file.read().splitlines()
labels0 = np.array([x.split() for x in lines], dtype=np.float32)
# If label file is empty x = np.array([x.split() for x in lines], dtype=np.float32)
if labels0.size is 0: if x.size is 0:
# Empty labels file
labels = np.array([]) labels = np.array([])
else: else:
# Normalized xywh to pixel xyxy format # Normalized xywh to pixel xyxy format
labels = labels0.copy() labels = x.copy()
labels[:, 1] = ratio * w * (labels0[:, 1] - labels0[:, 3] / 2) + padw labels[:, 1] = ratio * w * (x[:, 1] - x[:, 3] / 2) + padw
labels[:, 2] = ratio * h * (labels0[:, 2] - labels0[:, 4] / 2) + padh labels[:, 2] = ratio * h * (x[:, 2] - x[:, 4] / 2) + padh
labels[:, 3] = ratio * w * (labels0[:, 1] + labels0[:, 3] / 2) + padw labels[:, 3] = ratio * w * (x[:, 1] + x[:, 3] / 2) + padw
labels[:, 4] = ratio * h * (labels0[:, 2] + labels0[:, 4] / 2) + padh labels[:, 4] = ratio * h * (x[:, 2] + x[:, 4] / 2) + padh
else: else:
labels = np.array([]) labels = np.array([])