| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768 |
- import os
- import argparse
- import json
- def gen_rec_label(input_path, out_label):
- with open(out_label, "w") as out_file:
- with open(input_path, "r") as f:
- for line in f.readlines():
- tmp = line.strip("\n").replace(" ", "").split(",")
- img_path, label = tmp[0], tmp[1]
- label = label.replace('"', "")
- out_file.write(img_path + "\t" + label + "\n")
- def gen_det_label(root_path, input_dir, out_label):
- with open(out_label, "w") as out_file:
- for label_file in os.listdir(input_dir):
- img_path = os.path.join(root_path, label_file[3:-4] + ".jpg")
- label = []
- with open(
- os.path.join(input_dir, label_file), "r",
- encoding="utf-8-sig") as f:
- for line in f.readlines():
- tmp = line.strip("\n\r").replace("\xef\xbb\xbf",
- "").split(",")
- points = tmp[:8]
- s = []
- for i in range(0, len(points), 2):
- b = points[i:i + 2]
- b = [int(t) for t in b]
- s.append(b)
- result = {"transcription": tmp[8], "points": s}
- label.append(result)
- out_file.write(img_path + "\t" + json.dumps(
- label, ensure_ascii=False) + "\n")
- if __name__ == "__main__":
- parser = argparse.ArgumentParser()
- parser.add_argument(
- "--mode",
- type=str,
- default="rec",
- help="Generate rec_label or det_label, can be set rec or det", )
- parser.add_argument(
- "--root_path",
- type=str,
- default=".",
- help="The root directory of images.Only takes effect when mode=det ", )
- parser.add_argument(
- "--input_path",
- type=str,
- default=".",
- help="Input_label or input path to be converted", )
- parser.add_argument(
- "--output_label",
- type=str,
- default="out_label.txt",
- help="Output file name")
- args = parser.parse_args()
- if args.mode == "rec":
- print("Generate rec label")
- gen_rec_label(args.input_path, args.output_label)
- elif args.mode == "det":
- gen_det_label(args.root_path, args.input_path, args.output_label)
|
