From cb67d64a5f83facedef3d4a662703db326dae591 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fatih=20Baltac=C4=B1?= Date: Wed, 27 Mar 2019 17:44:19 +0300 Subject: [PATCH] Update datasets.py (#169) --- utils/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index be127210..5128bfe8 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -100,7 +100,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing assert len(self.img_files) > 0, 'No images found in %s' % path self.img_size = img_size self.augment = augment - self.label_files = [x.replace('images', 'labels').replace('.bmp', '.txt').replace('.jpg', '.txt') + self.label_files = [x.replace('images', 'labels').replace('.bmp', '.txt').replace('.jpg', '.txt').replace('.png', '.txt') for x in self.img_files] def __len__(self):