f.read().strip() (#1551)
This commit is contained in:
1 parent
9fa7f9f598
commit
cff9263490
1 file changed
+2
-2
+2
-2
@@ -450,7 +450,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
|
||||
if os.path.isfile(lb_file):
|
||||
nf += 1 # label found
|
||||
with open(lb_file, 'r') as f:
|
||||
l = np.array([x.split() for x in f.read().splitlines()], dtype=np.float32) # labels
|
||||
l = np.array([x.split() for x in f.read().strip().splitlines()], dtype=np.float32) # labels
|
||||
if len(l):
|
||||
assert l.shape[1] == 5, 'labels require 5 columns each'
|
||||
assert (l >= 0).all(), 'negative labels'
|
||||
@@ -897,7 +897,7 @@ def extract_boxes(path='../coco128/'): # from utils.datasets import *; extract_
|
||||
lb_file = Path(img2label_paths([str(im_file)])[0])
|
||||
if Path(lb_file).exists():
|
||||
with open(lb_file, 'r') as f:
|
||||
lb = np.array([x.split() for x in f.read().splitlines()], dtype=np.float32) # labels
|
||||
lb = np.array([x.split() for x in f.read().strip().splitlines()], dtype=np.float32) # labels
|
||||
|
||||
for j, x in enumerate(lb):
|
||||
c = int(x[0]) # class
|
||||
|
||||
Reference in new issue
Block a user