This commit is contained in:
Glenn Jocher 2019-05-05 21:49:04 +02:00
parent 439c5a9839
commit f05042b660
1 changed files with 1 additions and 1 deletions

View File

@ -130,7 +130,7 @@ class LoadWebcam: # for inference
class LoadImagesAndLabels(Dataset): # for training/testing
def __init__(self, path, img_size=416, batch_size=16, augment=False, rect=True):
def __init__(self, path, img_size=416, batch_size=16, augment=False, rect=False):
with open(path, 'r') as f:
img_files = f.read().splitlines()
self.img_files = list(filter(lambda x: len(x) > 0, img_files))