Merge pull request #2992 from livc/fix_flowers

fix flowers dataset bug

Merge pull request #2992 from livc/fix_flowers
fix flowers dataset bug
75ad5a6a · whs · GitHub · 155f727c · ea09edc9 · 75ad5a6a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/v2/dataset/flowers.py python/paddle/v2/dataset/flowers.py +1 -1

未找到文件。
--- 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)