From 9d922ebfe8559c4d0b44c4731cd5132f179d1323 Mon Sep 17 00:00:00 2001 From: Ryota Suwa Date: Sun, 20 Dec 2020 16:53:29 +0900 Subject: [PATCH] Fix directory path --- pytorch_networks/surface_normals/dataloader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_networks/surface_normals/dataloader.py b/pytorch_networks/surface_normals/dataloader.py index 43320f0..8452f72 100644 --- a/pytorch_networks/surface_normals/dataloader.py +++ b/pytorch_networks/surface_normals/dataloader.py @@ -160,7 +160,7 @@ def _create_lists_filenames(self, images_dir, labels_dir, masks_dir): numLabels = len(self._datalist_label) if numLabels == 0: - raise ValueError('No labels found in given directory. Searched for {}'.format(imageSearchStr)) + raise ValueError('No labels found in given directory. Searched for {}'.format(labels_dir)) if numImages != numLabels: raise ValueError('The number of images and labels do not match. Please check data,' + 'found {} images and {} labels in dirs:\n'.format(numImages, numLabels) + @@ -175,7 +175,7 @@ def _create_lists_filenames(self, images_dir, labels_dir, masks_dir): numMasks = len(self._datalist_mask) if numMasks == 0: - raise ValueError('No masks found in given directory. Searched for {}'.format(imageSearchStr)) + raise ValueError('No masks found in given directory. Searched for {}'.format(masks_dir)) if numImages != numMasks: raise ValueError('The number of images and masks do not match. Please check data,' + 'found {} images and {} masks in dirs:\n'.format(numImages, numMasks) +