convert csv.py
2.38 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
import os
import csv
import xlsxwriter
from PIL import Image
import re
# make empty csv file
workbook = xlsxwriter.Workbook("dataset.xlsx")
worksheet = workbook.add_worksheet("sheet1")
# resize col
worksheet.set_column("A:A", 60)
# get current directory path
directory = os.getcwd()
img_file_name = []
text_file_name = []
for filename in os.listdir(directory):
if filename.endswith(".png"):
path = os.path.join(directory, filename)
img_file_name.append(path)
elif filename.endswith(".txt"):
path = os.path.join(directory, filename)
text_file_name.append(path)
print(img_file_name) # for check
print(text_file_name) # for check
# text annotation
text = []
for t in text_file_name:
plate = []
with open(t, "r") as f:
for line in f:
if line[:5] == "plate":
plate.append(line.strip("plate: ").strip())
elif line[:14] == "position_plate":
numbers = list(map(int, line.strip("position_plate: ").strip().split()))
plate.append(numbers)
elif "char" in line:
numbers = re.findall("\d+", line[8:])
plate.append(list(map(int, numbers)))
text.append(plate)
print(text) # for check
#################################################################################
#################### error : 마지막 image, text에 대해서만 됨 ####################
#################################################################################
# insert to csv
idx = 0
# for filename in os.listdir(directory):
for filename in img_file_name:
# if filename.endswith(".png"):
# col A, B
for i in range(len(text[idx][0]) - 1):
A = "A" + str(i + 1)
path = os.path.join(directory, filename)
worksheet.write(A, path)
B = "B" + str(i + 1)
worksheet.write(B, text[idx][0])
idx += 1
for filename in text_file_name:
# if filename.endswith(".txt"):
# col C, D, E, F : x1, y1, x2, y2
for i in range(len(text[0]) - 2):
C = "C" + str(i + 1)
worksheet.write(C, text[idx][i + 2][0])
D = "D" + str(i + 1)
worksheet.write(D, text[idx][i + 2][1])
E = "E" + str(i + 1)
worksheet.write(E, text[idx][i + 2][0] + text[idx][i + 2][2])
F = "F" + str(i + 1)
worksheet.write(F, text[idx][i + 2][1] + text[idx][i + 2][3])
idx += 1
workbook.close()