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

updates

This commit is contained in:
Glenn Jocher 2019-11-12 14:16:54 -08:00
parent 579fdc57f8
commit 470ef6bc92
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
utils/datasets.py
View File

@ -579,7 +579,6 @@ def load_mosaic(self, index):
with open(label_path, 'r') as f: with open(label_path, 'r') as f:
x = np.array([x.split() for x in f.read().splitlines()], dtype=np.float32) x = np.array([x.split() for x in f.read().splitlines()], dtype=np.float32)
labels = []
if x.size > 0: if x.size > 0:
# Normalized xywh to pixel xyxy format # Normalized xywh to pixel xyxy format
labels = x.copy() labels = x.copy()
@ -587,6 +586,8 @@ def load_mosaic(self, index):
labels[:, 2] = h * (x[:, 2] - x[:, 4] / 2) + padh labels[:, 2] = h * (x[:, 2] - x[:, 4] / 2) + padh
labels[:, 3] = w * (x[:, 1] + x[:, 3] / 2) + padw labels[:, 3] = w * (x[:, 1] + x[:, 3] / 2) + padw
labels[:, 4] = h * (x[:, 2] + x[:, 4] / 2) + padh labels[:, 4] = h * (x[:, 2] + x[:, 4] / 2) + padh
else:
labels = np.zeros((0, 5), dtype=np.float32)
labels4.append(labels) labels4.append(labels)
if len(labels4): if len(labels4):