From 225950720250ecd0e6ac113d1a11d424ea2575ea Mon Sep 17 00:00:00 2001
From: "shaohua.zhang" <BeyondYourself@users.noreply.github.com>
Date: Mon, 31 Aug 2020 13:50:56 +0800
Subject: [PATCH] add the img path

---
 ppocr/data/rec/dataset_traversal.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ppocr/data/rec/dataset_traversal.py b/ppocr/data/rec/dataset_traversal.py
index 84f325b9..06365ec1 100755
--- a/ppocr/data/rec/dataset_traversal.py
+++ b/ppocr/data/rec/dataset_traversal.py
@@ -131,6 +131,7 @@ class LMDBReader(object):
                     else:
                         norm_img = process_image(
                             img=img,
+                            img_path=single_img,
                             image_shape=self.image_shape,
                             char_ops=self.char_ops,
                             tps=self.use_tps,
@@ -168,6 +169,7 @@ class LMDBReader(object):
                             else:
                                 outs = process_image(
                                     img=img,
+                                    img_path=None,
                                     image_shape=self.image_shape,
                                     label=label,
                                     char_ops=self.char_ops,
@@ -263,6 +265,7 @@ class SimpleReader(object):
                     else:
                         norm_img = process_image(
                             img=img,
+                            img_path=single_img,
                             image_shape=self.image_shape,
                             char_ops=self.char_ops,
                             tps=self.use_tps,
@@ -309,6 +312,7 @@ class SimpleReader(object):
                     else:
                         outs = process_image(
                             img=img,
+                            img_path=img_path,
                             image_shape=self.image_shape,
                             label=label,
                             char_ops=self.char_ops,
-- 
GitLab