diff options
| -rw-r--r-- | Code/test.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/Code/test.py b/Code/test.py index 134b8b9..9072c2b 100644 --- a/Code/test.py +++ b/Code/test.py @@ -2,7 +2,8 @@ import numpy as np import constants as c from utils import normalize_frames, get_test_batch from glob import glob -from scipy.misc import imread, imsave +from scipy.ndimage import imread +from scipy.misc import imsave import os def save_batch(batch, num_rec_out): |