-
Notifications
You must be signed in to change notification settings - Fork 4
/
annotation_plugin.py
64 lines (49 loc) · 1.34 KB
/
annotation_plugin.py
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
import sys
if __name__ == '__main__':
fin = open(sys.argv[1], 'r')
fout = open('./annotation_plugin_grammar.txt', 'w')
annotation_num = int(fin.readline().rstrip('\n'))
print annotation_num
annotation_count = 0
while annotation_count < annotation_num:
fin.readline();
annotation_count += 1
fin.readline();
sample_num = int(fin.readline().rstrip('\n'))
print sample_num
sample_count = 0;
while sample_count < sample_num:
sample = fin.readline().rstrip('\n')
left_parts = []
left_parts = sample.split(' ] ')
for part in left_parts:
left_bracket = part.find('[')
annotation = part[0 : left_bracket].rstrip(' ')
if annotation == '0':
fout.write('approach ')
elif annotation == '1':
fout.write('move ')
elif annotation == '2':
fout.write('grasp_left ')
elif annotation == '3':
fout.write('grasp_right ')
elif annotation == '4':
fout.write('ungrasp_left ')
elif annotation == '5':
fout.write('ungrasp_right ')
elif annotation == '6':
fout.write('twist ')
elif annotation == '7':
fout.write('push ')
elif annotation == '8':
fout.write('neutral ')
elif annotation == '9':
fout.write('pull ')
elif annotation == '10':
fout.write('pinch ')
elif annotation == '11':
fout.write('unpinch ')
fout.write('\n\n');
sample_count += 1
fin.close()
fout.close()