From db0841964f57962ce736c598485496f03f2da6ca Mon Sep 17 00:00:00 2001
From: Luc Giffon <luc.giffon@lis-lab.fr>
Date: Sun, 2 Sep 2018 10:46:27 +0200
Subject: [PATCH] test resize avec check des nouvelles tailles
---
skluc/data/mldatasets/ImageDataset.py | 1 +
skluc/test/test_transformation/TestResizeTransformer.py | 3 +++
2 files changed, 4 insertions(+)
diff --git a/skluc/data/mldatasets/ImageDataset.py b/skluc/data/mldatasets/ImageDataset.py
index b8f6ee6..904f4fd 100644
--- a/skluc/data/mldatasets/ImageDataset.py
+++ b/skluc/data/mldatasets/ImageDataset.py
@@ -79,6 +79,7 @@ class ImageDataset(Dataset):
length_by_chanel = int(length_by_chanel)
images_mat = np.reshape(images_vec, (images_vec.shape[0], length_by_chanel, self.DEPTH), order='F')
images = np.reshape(images_mat, (images_mat.shape[0], self.HEIGHT, self.WIDTH, self.DEPTH), order='C')
+ logger.debug("Images mat shape: {}".format(images.shape))
setattr(self, kw, LabeledData(images, labels))
def flatten(self):
diff --git a/skluc/test/test_transformation/TestResizeTransformer.py b/skluc/test/test_transformation/TestResizeTransformer.py
index 60ea6c0..23ba871 100644
--- a/skluc/test/test_transformation/TestResizeTransformer.py
+++ b/skluc/test/test_transformation/TestResizeTransformer.py
@@ -33,6 +33,9 @@ class TestResizeTransformer(unittest.TestCase):
d.to_image()
trans = ResizeTransformer(data_name=data_name, output_shape=size)
d.apply_transformer(transformer=trans)
+ new_size = d.train.data[0].shape[:-1]
+ self.assertEqual(new_size[0], size[0])
+ self.assertEqual(new_size[1], size[1])
del trans
def test_init(self):
--
GitLab