Image2Image resize at test time. fix #169

60200fc1 · Yuxin Wu · aa673e97 · 60200fc1
Commit 60200fc1 authored Mar 02, 2017 by Yuxin Wu
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

examples/GAN/Image2Image.py examples/GAN/Image2Image.py +3 -1

No files found.
--- a/examples/GAN/Image2Image.py
+++ b/examples/GAN/Image2Image.py
@@ -191,7 +191,9 @@ def sample(datadir, model_path):

    imgs = glob.glob(os.path.join(datadir, '*.jpg'))
    ds = ImageFromFile(imgs, channel=3, shuffle=True)
-    ds = BatchData(MapData(ds, lambda dp: split_input(dp[0])), 6)
+    ds = MapData(ds, lambda dp: split_input(dp[0]))
+    ds = AugmentImageComponents(ds, [imgaug.Resize(256)], (0, 1))
+    ds = BatchData(ds, 6)

    pred = SimpleDatasetPredictor(pred, ds)
    for o in pred.get_result():