From 0957603edd1ba9aee7490ff72ddce61307835557 Mon Sep 17 00:00:00 2001 From: Andrew Hundt Date: Tue, 19 Dec 2017 12:04:23 -0500 Subject: [PATCH] test_image_segmentation.py correct import --- tests/keras_contrib/preprocessing/test_image_segmentation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/keras_contrib/preprocessing/test_image_segmentation.py b/tests/keras_contrib/preprocessing/test_image_segmentation.py index b85d899..5a0bb13 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 keras_contrib.utils import SegDataGenerator +from keras_contrib.preprocessing.image_segmentation import SegDataGenerator from PIL import Image as PILImage import numpy as np