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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
| {
| "cells": [
| {
| "cell_type": "code",
| "execution_count": null,
| "metadata": {
| "collapsed": true
| },
| "outputs": [],
| "source": []
| },
| {
| "cell_type": "code",
| "execution_count": 27,
| "metadata": {
| "collapsed": true
| },
| "outputs": [],
| "source": [
| "AFLW2000 = '/Data/nruiz9/data/facial_landmarks/AFLW2000/'\n",
| "AFLW = '/Data/nruiz9/data/facial_landmarks/AFLW/aflw_cropped_loose/'"
| ]
| },
| {
| "cell_type": "code",
| "execution_count": 34,
| "metadata": {
| "collapsed": true
| },
| "outputs": [],
| "source": [
| "list_2000 = '/Data/nruiz9/data/facial_landmarks/AFLW/KEPLER_test_split.txt'\n",
| "list_normal = '/Data/nruiz9/data/facial_landmarks/AFLW/aflw_cropped_loose/filename_list_filtered.txt'"
| ]
| },
| {
| "cell_type": "code",
| "execution_count": 35,
| "metadata": {
| "collapsed": false
| },
| "outputs": [
| {
| "name": "stdout",
| "output_type": "stream",
| "text": [
| "961\n"
| ]
| }
| ],
| "source": [
| "fid = open(list_2000, 'r')\n",
| "dict_2000 = dict()\n",
| "for line in fid:\n",
| " line = line.strip('\\n').split('/')\n",
| " dict_2000[line[-1]] = 1\n",
| "print len(dict_2000)\n",
| "fid.close()"
| ]
| },
| {
| "cell_type": "code",
| "execution_count": 36,
| "metadata": {
| "collapsed": false
| },
| "outputs": [
| {
| "name": "stdout",
| "output_type": "stream",
| "text": [
| "943 19537\n"
| ]
| }
| ],
| "source": [
| "fid = open(list_normal, 'r')\n",
| "naked = list_normal.strip('.txt')\n",
| "train = open(naked + '_train.txt', 'wb')\n",
| "test = open(naked + '_test.txt', 'wb')\n",
| "test_dict = dict()\n",
| "train_dict = dict()\n",
| "for line in fid:\n",
| " line = line.strip('\\n')\n",
| " name = line.split('/')[-1]\n",
| " if name+'.jpg' in dict_2000.keys():\n",
| " test.write(line + '\\n')\n",
| " test_dict[line] = 1\n",
| " else:\n",
| " train.write(line + '\\n')\n",
| " train_dict[line] = 1\n",
| "\n",
| "print len(test_dict), len(train_dict)"
| ]
| },
| {
| "cell_type": "code",
| "execution_count": null,
| "metadata": {
| "collapsed": true
| },
| "outputs": [],
| "source": []
| }
| ],
| "metadata": {
| "anaconda-cloud": {},
| "kernelspec": {
| "display_name": "Python [conda root]",
| "language": "python",
| "name": "conda-root-py"
| },
| "language_info": {
| "codemirror_mode": {
| "name": "ipython",
| "version": 2
| },
| "file_extension": ".py",
| "mimetype": "text/x-python",
| "name": "python",
| "nbconvert_exporter": "python",
| "pygments_lexer": "ipython2",
| "version": "2.7.12"
| }
| },
| "nbformat": 4,
| "nbformat_minor": 1
| }
|
|
