bug fix

cfdb4b57 · Yuxin Wu · d7a020fc · cfdb4b57
Commit cfdb4b57 authored May 10, 2016 by Yuxin Wu
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/tensorpack/dataflow/image.py
+++ b/tensorpack/dataflow/image.py
@@ -49,7 +49,7 @@ class AugmentImageComponent(MapDataComponent):
        """
        self.augs = AugmentorList(augmentors)
        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):
        self.ds.reset_state()