Skip to content

S
seminar-breakout
Commit ae84b52a authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

bugfix in ImageFromFile

parent 8a3d0d60
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
tensorpack/dataflow/image.py
View file @ ae84b52a
...@@ -27,7 +27,9 @@ class ImageFromFile(RNGDataFlow): ...@@ -27,7 +27,9 @@ class ImageFromFile(RNGDataFlow):
self.files = files self.files = files
self.channel = int(channel) self.channel = int(channel)
self.imread_mode = cv2.IMREAD_GRAYSCALE if self.channel == 1 else cv2.IMREAD_COLOR 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 self.shuffle = shuffle
def size(self): def size(self):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment