Bug fix in dataset loading

sergionr2 · Jan 8, 2018 · badaafd · badaafd
1 parent 751e7cf
commit badaafd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/train/utils.py b/train/utils.py
@@ -74,6 +74,8 @@ def loadDataset(split_seed=42, folder='', split=True, augmented=True):
         y[idx] = x_center / width
 
         image_path = '{}/{}.jpg'.format(folder, images_dict[name]['output_name'])
+        path = images_dict[name]['output_name']
+        image_path = '{}/{}.jpg'.format(folder, path)
         im = cv2.imread(image_path)
         # Resize and normalize input
         X[idx, :] = preprocessImage(im, WIDTH, HEIGHT)