bugfix in ImageFromFile

ae84b52a · Yuxin Wu · 8a3d0d60 · ae84b52a
Commit ae84b52a authored May 02, 2017 by Yuxin Wu
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

tensorpack/dataflow/image.py tensorpack/dataflow/image.py +3 -1

No files found.
--- a/tensorpack/dataflow/image.py
+++ b/tensorpack/dataflow/image.py
@@ -27,7 +27,9 @@ class ImageFromFile(RNGDataFlow):
        self.files = files
        self.channel = int(channel)
        self.imread_mode = cv2.IMREAD_GRAYSCALE if self.channel == 1 else cv2.IMREAD_COLOR
-        self.resize = shape2d(resize)
+        if resize is not None:
+            resize = shape2d(resize)
+        self.resize = resize
        self.shuffle = shuffle

    def size(self):