-
Notifications
You must be signed in to change notification settings - Fork 0
/
correct_casia1.py
43 lines (34 loc) · 1.02 KB
/
correct_casia1.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
import os
import re
from glob import glob
anno = glob('Annotations/*')
img = glob('JPEGImages/*')
list1 = []
dict1 = {}
for a in anno:
aa = re.split('[/.]', a)[1] # Sp_D_NRN_A_cha0036_art0090_0465
list1.append(aa)
s = re.split('[/_.]', a)[1:-1]
dict1[aa] = s
# data1_path = '/home/duxiaowey/my_ps/src/faster-rcnn/data/CASIA/CASIA1/Annotations'
# data2_path = '/home/duxiaowey/my_ps/src/faster-rcnn/data/mycasia/JPEGImages'
# file1_list = os.listdir(data1_path)
# file2_list = os.listdir(data2_path)
list2 = []
dict2 = {}
for i in img:
ii = re.split('[/.]', i)[1] # Sp_D_NRN_A_cha0036_art0090_0465
list2.append(ii)
s = re.split('[/_.]', i)[1:-1]
dict2[ii] = s # ['Sp', 'S', 'NNN', 'A', 'arc0051', 'arc0029', '0018']
'''删除anno字典中与img相同的元素'''
for f in list1:
if f in list2:
del dict1[f]
'''删除img字典中与anno相同的元素'''
for f in list2:
if f in list1:
del dict2[f]
cha = set(list1).difference(set(list2))
print(cha)
print(len(dict2))