forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix gen label

This commit is contained in:
littletomatodonkey 2020-09-23 11:03:45 +00:00
parent 3a18b08fc5
commit 43f6ab7a63
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
train_data/gen_label.py
View File

@ -13,6 +13,7 @@
#limitations under the License. #limitations under the License.
import os import os
import argparse import argparse
import json
def gen_rec_label(input_path, out_label): def gen_rec_label(input_path, out_label):
@ -32,15 +33,19 @@ def gen_det_label(root_path, input_dir, out_label):
label = [] label = []
with open(os.path.join(input_dir, label_file), 'r') as f: with open(os.path.join(input_dir, label_file), 'r') as f:
for line in f.readlines(): for line in f.readlines():
tmp = line.strip("\n\r").replace("\xef\xbb\xbf", "").split(',') tmp = line.strip("\n\r").replace("\xef\xbb\xbf",
points = tmp[:-2] "").split(',')
points = tmp[:8]
s = [] s = []
for i in range(0, len(points), 2): for i in range(0, len(points), 2):
b = points[i:i + 2] b = points[i:i + 2]
b = [int(t) for t in b]
s.append(b) s.append(b)
result = {"transcription": tmp[-1], "points": s} result = {"transcription": tmp[8], "points": s}
label.append(result) label.append(result)
out_file.write(img_path + '\t' + str(label) + '\n')
out_file.write(img_path + '\t' + json.dumps(
label, ensure_ascii=False) + '\n')
if __name__ == "__main__": if __name__ == "__main__":