-
Notifications
You must be signed in to change notification settings - Fork 1
/
AgeGenCom.py
59 lines (50 loc) · 1.49 KB
/
AgeGenCom.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
import numpy as np
import os
import linecache as lc
#filelist ='/home/miaoqianwen/AgePre/detTrain'
#filelist ='/home/miaoqianwen/AgePre/detTest'
filelist ='/home/miaoqianwen/AgePre/detVal'
dicimdb = '/home/SSD1/AgeDataBase/imdbAge/imdb_dem'
dicwiki = '/home/SSD1/AgeDataBase/wikiAge/wiki_dem'
arr = []
with open(dicimdb) as lmfile:
lineNum=sum(1 for _ in lmfile)
it=iter(range(1, lineNum))
for m in it:
line = lc.getline(dicimdb, m)
line = line.rstrip('\n')
file = line.split(' ')
Gender = file[1]
file = file[0].split('/')
ImgName = '/home/SSD1/AgeDataBase/imdbAge/imdbMTCNN/' + file[1]
arr.append(ImgName)
arr.append(Gender)
with open(dicwiki) as lmfile:
lineNum=sum(1 for _ in lmfile)
it=iter(range(1, lineNum))
for m in it:
line = lc.getline(dicwiki, m)
line = line.rstrip('\n')
file = line.split(' ')
Gender = file[1]
file = file[0].split('/')
ImgName = '/home/SSD1/AgeDataBase/wikiAge/wikiMTCNN/' + file[1]
arr.append(ImgName)
arr.append(Gender)
with open(filelist) as lmfile:
lineNum=sum(1 for _ in lmfile)
it=iter(range(1, lineNum))
for m in it:
line = lc.getline(filelist, m)
line = line.rstrip('\n')
file = line.split(' ')
ImgName = file[0]
try:
idx = arr.index(ImgName)
gender = arr [idx+1]
iAge = file[1]
newline = ImgName + " " + iAge + " " + gender
with open('detValG',"a+") as outfile:
outfile.write(newline + "\n")
except:
print(ImgName)