diff --git a/ppocr/data/det/east_process.py b/ppocr/data/det/east_process.py index f9a07440..e2581caa 100755 --- a/ppocr/data/det/east_process.py +++ b/ppocr/data/det/east_process.py @@ -17,6 +17,7 @@ import cv2 import numpy as np import json import sys +import os class EASTProcessTrain(object): def __init__(self, params): @@ -52,7 +53,7 @@ class EASTProcessTrain(object): label_infor = label_infor.decode() label_infor = label_infor.encode('utf-8').decode('utf-8-sig') substr = label_infor.strip("\n").split("\t") - img_path = self.img_set_dir + substr[0] + img_path = os.path.join(self.img_set_dir, substr[0]) label = json.loads(substr[1]) nBox = len(label) wordBBs, txts, txt_tags = [], [], []