From 7bc0d8288fd38473fbd56548c33c23436a4d3442 Mon Sep 17 00:00:00 2001 From: Andrew Hundt Date: Tue, 19 Dec 2017 11:58:34 -0500 Subject: [PATCH] test_image_segmentation.py init SegDataGenerator() --- .../keras_contrib/preprocessing/test_image_segmentation.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/keras_contrib/preprocessing/test_image_segmentation.py b/tests/keras_contrib/preprocessing/test_image_segmentation.py index 0c5fe0e..b85d899 100644 --- a/tests/keras_contrib/preprocessing/test_image_segmentation.py +++ b/tests/keras_contrib/preprocessing/test_image_segmentation.py @@ -1,5 +1,5 @@ from keras.preprocessing.image import img_to_array, array_to_img -from utils import SegDataGenerator +from keras_contrib.utils import SegDataGenerator from PIL import Image as PILImage import numpy as np @@ -46,3 +46,7 @@ test_random_crop = lambda: test_crop(SegDataGenerator.random_crop) test_pair_center_crop = lambda: test_pair_crop(SegDataGenerator.pair_center_crop) test_pair_random_crop = lambda: test_pair_crop(SegDataGenerator.pair_random_crop) + + +def test_seg_data_generator(): + datagen = SegDataGenerator()