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

Empty label file may cause index error

This commit is contained in:
WannaSeaU 2019-03-22 18:59:09 +08:00 committed by GitHub
parent 476724be2d
commit cd188dbde6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
utils/datasets.py
View File

@ -141,7 +141,10 @@ class LoadImagesAndLabels(Dataset): # for training/testing
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) labels0 = np.array([x.split() for x in lines], dtype=np.float32)
# If label file is empty
if labels0.size is 0:
labels = np.array([])
else:
# Normalized xywh to pixel xyxy format # Normalized xywh to pixel xyxy format
labels = labels0.copy() labels = labels0.copy()
labels[:, 1] = ratio * w * (labels0[:, 1] - labels0[:, 3] / 2) + padw labels[:, 1] = ratio * w * (labels0[:, 1] - labels0[:, 3] / 2) + padw