test_image_segmentation.py correct import

This commit is contained in:
Andrew Hundt
2017-12-19 12:04:23 -05:00
committed by GitHub
parent 7bc0d8288f
commit 0957603edd
@@ -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