Merge pull request #2992 from livc/fix_flowers

fix flowers dataset bug
8 years ago · 75ad5a6a55
parent 155f727c21 ea09edc991
commit 75ad5a6a55
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/v2/dataset/flowers.py
+++ b/python/paddle/v2/dataset/flowers.py
@ -116,7 +116,7 @@ def reader_creator(data_file,
            data = batch['data']
            labels = batch['label']
            for sample, label in itertools.izip(data, batch['label']):
-                yield sample, int(label)
+                yield sample, int(label) - 1

    if use_xmap:
        return xmap_readers(mapper, reader, cpu_count(), buffered_size)