From fb69843af5abd71ecab7ed8958127dd5c19a8091 Mon Sep 17 00:00:00 2001
From: Luc Giffon <luc.giffon@lis-lab.fr>
Date: Sun, 2 Sep 2018 12:19:03 +0200
Subject: [PATCH] test vgg19 transformer working
---
.../test_transformation/TestVGG19Transformer.py | 10 +++-------
1 file changed, 3 insertions(+), 7 deletions(-)
diff --git a/skluc/test/test_data/test_transformation/TestVGG19Transformer.py b/skluc/test/test_data/test_transformation/TestVGG19Transformer.py
index ce3c33a..d5aca71 100644
--- a/skluc/test/test_data/test_transformation/TestVGG19Transformer.py
+++ b/skluc/test/test_data/test_transformation/TestVGG19Transformer.py
@@ -1,24 +1,19 @@
import unittest
-from skluc.main.data.mldatasets.Cifar100FineDataset import Cifar100FineDataset
from skluc.main.data.mldatasets.Cifar10Dataset import Cifar10Dataset
-from skluc.main.data.mldatasets.SVHNDataset import SVHNDataset
+
from skluc.main.data.transformation.VGG19Transformer import VGG19Transformer
+
from skluc.main.utils import logger
class TestVGG19Transformer(unittest.TestCase):
def setUp(self):
self.lst_name_cut_layers = [
- "block5_conv4",
- "block4_conv4",
- "block5_pool",
"block3_pool"
]
self.dict_datasets = {
"cifar10": Cifar10Dataset,
- "cifar100": Cifar100FineDataset,
- "svhn": SVHNDataset
}
def test_transform(self):
@@ -45,6 +40,7 @@ class TestVGG19Transformer(unittest.TestCase):
for name_cut_layer in self.lst_name_cut_layers:
logger.info("Testing cut layer {}".format(name_cut_layer))
trans = VGG19Transformer(data_name=data_name, cut_layer_name=name_cut_layer)
+ logger.debug(trans.keras_model)
del trans
--
GitLab