wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-09-11 12:20:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

test_image_segmentation.py init SegDataGenerator()

Browse Source
This commit is contained in:
Andrew Hundt
2017-12-19 11:58:34 -05:00
committed by GitHub
parent 09f967889c
commit 7bc0d8288f
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/keras_contrib/preprocessing/test_image_segmentation.py
+5 -1
View File
@@ -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()