Skip to content

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

bug fix

parent d7a020fc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/dataflow/image.py
View file @ cfdb4b57
...@@ -49,7 +49,7 @@ class AugmentImageComponent(MapDataComponent): ...@@ -49,7 +49,7 @@ class AugmentImageComponent(MapDataComponent):
""" """
self.augs = AugmentorList(augmentors) self.augs = AugmentorList(augmentors)
super(AugmentImageComponent, self).__init__( super(AugmentImageComponent, self).__init__(
ds, lambda x: self.augs.augment(Image(x)).arr, index) ds, lambda x: self.augs.augment(x), index)
def reset_state(self): def reset_state(self):
self.ds.reset_state() self.ds.reset_state()
......
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