未验证 提交 c12e9ec5 编写于 作者: S shaohua.zhang 提交者: GitHub

Update east_process.py

上级 08055099
...@@ -17,6 +17,7 @@ import cv2 ...@@ -17,6 +17,7 @@ import cv2
import numpy as np import numpy as np
import json import json
import sys import sys
import os
class EASTProcessTrain(object): class EASTProcessTrain(object):
def __init__(self, params): def __init__(self, params):
...@@ -52,7 +53,7 @@ class EASTProcessTrain(object): ...@@ -52,7 +53,7 @@ class EASTProcessTrain(object):
label_infor = label_infor.decode() label_infor = label_infor.decode()
label_infor = label_infor.encode('utf-8').decode('utf-8-sig') label_infor = label_infor.encode('utf-8').decode('utf-8-sig')
substr = label_infor.strip("\n").split("\t") 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]) label = json.loads(substr[1])
nBox = len(label) nBox = len(label)
wordBBs, txts, txt_tags = [], [], [] wordBBs, txts, txt_tags = [], [], []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册