From b965b6e9b7927a391974f329bd1ac7fac9649792 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 26 Aug 2018 11:52:27 +0200 Subject: [PATCH] updates --- utils/datasets.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index 50af4d78..d6036d15 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -2,6 +2,7 @@ import glob import math import os import random +from sys import platform import cv2 import numpy as np @@ -63,7 +64,11 @@ class ListDataset(): # for training #self.img_files = sorted(glob.glob('%s/*.*' % path)) with open(path, 'r') as file: self.img_files = file.readlines() - self.img_files = [path.replace('\n', '').replace('/images','/Users/glennjocher/Downloads/DATA/coco/images') for path in self.img_files] + + if platform == 'darwin': # macos + self.img_files = [path.replace('\n', '').replace('/images','/Users/glennjocher/Downloads/DATA/coco/images') for path in self.img_files] + else: + self.img_files = [path.replace('\n', '').replace('/images','../coco/images') for path in self.img_files] self.label_files = [path.replace('images', 'labels').replace('.png', '.txt').replace('.jpg', '.txt') for path in self.img_files]