Peter M. Groen / openbr

Commit 8c5153cbc5f81667f370c29b2faca6f6bd587e32

Authored by Josh Klontz
1 parent f722baa3

added more attributes, training takes pretty long now but accuracy is much improved

Inline Side-by-side
Showing 1 changed file with 38 additions and 3 deletions
scripts/attributes.sh
  View file @8c5153c
... ... @@ -20,13 +20,22 @@ DATA=${DATA:-~/data/CUHK-VHDC}
20 20 if [ ! -f $ATTDIR/all.model ]; then
21 21 br -crossValidate 2 -algorithm "CrossValidate($BASE+ \
22 22 ($BROW+ \
  23 + TurkClassifier(eyebrowposition,[closebrows,highbrows],3)/ \
23 24 TurkClassifier(unibrow,[unibrow],3)/ \
24 25 TurkClassifier(eyebroworientation,[eyebrowsdown,eyebrowsuptodown],3)/ \
25 26 TurkClassifier(thickeyebrows,[thickeyebrows,lighteyebrows],3))/ \
26 27 ($MOUTH+ \
27 28 TurkClassifier(smiling,[smiling],3)/ \
  29 + TurkClassifier(lipthickness,[cherry,big,small],3)/ \
  30 + TurkClassifier(mouthbite,[underbite,overbite],3)/ \
  31 + TurkClassifier(mouthopen,[closed,noteeth,halfteeth,allteeth],3)/ \
  32 + TurkClassifier(mouthwidth,[small,wide],3)/ \
  33 + TurkClassifier(mustache,[nomustache,linemustache,lightmustache,normalmustache,down],3)/ \
28 34 TurkClassifier(mouthasymmetry,[asymmetrical],3))/ \
29 35 ($EYES+ \
  36 + TurkClassifier(eyeseparation,[close,wide],3)/ \
  37 + TurkClassifier(eyeslant,[slant2,slant1,wild],3)/ \
  38 + TurkClassifier(benteyes,[bent])/ \
30 39 TurkClassifier(eyecolor,[darkeyes,lighteyes],3)/ \
31 40 TurkClassifier(baggyeyes,[baggy],3)/ \
32 41 TurkClassifier(almondeyes,[almond],3)/ \
... ... @@ -34,32 +43,49 @@ if [ ! -f $ATTDIR/all.model ]; then
34 43 TurkClassifier(sleepyeyes,[sleepy],3)/ \
35 44 TurkClassifier(lineeyes,[line],3)/ \
36 45 TurkClassifier(roundeyes,[round],3)/ \
  46 + TurkClassifier(sharpeyes,[sharp],3)/ \
37 47 TurkClassifier(smalleyes,[smalleyes],3)/ \
38 48 TurkClassifier(glasses,[glasses],3)/ \
39 49 TurkClassifier(eyelashvisibility,[feweyelashes],3))/ \
40 50 ($FACE+ \
41 51 TurkClassifier(gender,[male],3)/ \
  52 + TurkClassifier(faceshape,[round,triangular,rectangular],3)/ \
42 53 TurkClassifier(cheekdensity,[puffy,in,normal],3)/ \
43 54 TurkClassifier(facemarks,[scars,moles,normal],3)/ \
44 55 TurkClassifier(facelength,[long],3)/ \
  56 + TurkClassifier(nosetoeyedist,[short,long],3)/ \
45 57 TurkClassifier(nosetomouthdist,[long,small],3))/ \
46 58 ($HAIR+ \
47 59 TurkClassifier(foreheadwrinkles,[wrinkled],3)/ \
  60 + TurkClassifier(foreheadsize,[smallforehead,largeforehead],3)/ \
48 61 TurkClassifier(haircolor,[darkhair,lighthair,greyhair],3)/ \
  62 + TurkClassifier(hairdensity,[thick,bald,thin,halfbald],3)/ \
  63 + TurkClassifier(widowspeak,[widowspeak],3)/ \
49 64 TurkClassifier(hairstyle,[curlyhair],3))/ \
50 65 ($NOSE+ \
51 66 TurkClassifier(noseorientation,[upnose,downnose],3)/ \
52 67 TurkClassifier(nosewidth,[small,thick],3)/ \
53   - TurkClassifier(nosesize,[smallnose,bignose],3))/ \
  68 + TurkClassifier(nosesize,[smallnose,bignose],3)/ \
  69 + TurkClassifier(brokennose,[broken],3))/ \
54 70 ($JAW+ \
  71 + TurkClassifier(beard,[nobeard,bigbeard,lightbeard,goatee,linebeard,normalbeard,lincolnbeard],3)/ \
55 72 TurkClassifier(chinsize,[shortchin,longchin],3)) \
56 73 ): \
57 74 CrossValidate+Fuse([ \
  75 +Turk(eyebrowposition,[closebrows,highbrows],3), \
58 76 Turk(unibrow,[unibrow],3), \
59 77 Turk(eyebroworientation,[eyebrowsdown,eyebrowsuptodown],3), \
60 78 Turk(thickeyebrows,[thickeyebrows,lighteyebrows],3), \
61 79 Turk(smiling,[smiling],3), \
  80 +Turk(lipthickness,[cherry,big,small],3), \
  81 +Turk(mouthbite,[underbite,overbite],3), \
  82 +Turk(mouthopen,[closed,noteeth,halfteeth,allteeth],3), \
  83 +Turk(mouthwidth,[small,wide],3), \
  84 +Turk(mustache,[nomustache,linemustache,lightmustache,normalmustache,down],3), \
62 85 Turk(mouthasymmetry,[asymmetrical],3), \
  86 +Turk(eyeseparation,[close,wide],3), \
  87 +Turk(eyeslant,[slant2,slant1,wild],3), \
  88 +Turk(benteyes,[bent],3), \
63 89 Turk(eyecolor,[darkeyes,lighteyes],3), \
64 90 Turk(baggyeyes,[baggy],3), \
65 91 Turk(almondeyes,[almond],3), \
... ... @@ -67,19 +93,29 @@ Turk(buriedeyes,[buriedeyes],3), \
67 93 Turk(sleepyeyes,[sleepy],3), \
68 94 Turk(lineeyes,[line],3), \
69 95 Turk(roundeyes,[round],3), \
  96 +Turk(sharpeyes,[sharp],3), \
70 97 Turk(smalleyes,[smalleyes],3), \
71 98 Turk(glasses,[glasses],3), \
  99 +Turk(eyelashvisibility,[feweyelashes],3), \
  100 +Turk(gender,[male],3), \
  101 +Turk(faceshape,[round,triangular,rectangular],3), \
72 102 Turk(cheekdensity,[puffy,in,normal],3), \
73 103 Turk(facemarks,[scars,moles,normal],3), \
74 104 Turk(facelength,[long],3), \
  105 +Turk(nosetoeyedist,[short,long],3), \
75 106 Turk(nosetomouthdist,[long,small],3), \
76 107 Turk(foreheadwrinkles,[wrinkled],3), \
  108 +Turk(foreheadsize,[smallforehead,largeforehead],3), \
77 109 Turk(haircolor,[darkhair,lighthair,greyhair],3), \
  110 +Turk(hairdensity,[thick,bald,thin,halfbald],3), \
  111 +Turk(widowspeak,[widowspeak],3), \
78 112 Turk(hairstyle,[curlyhair],3), \
79 113 Turk(noseorientation,[upnose,downnose],3), \
80 114 Turk(nosewidth,[small,thick],3), \
81 115 Turk(nosesize,[smallnose,bignose],3), \
82   -Turk(chinsize,[shortchin,longchin],3)],indices=[0,1,2,3,4,5,6,7,8,9,10,11,12,13,16,17,18,19,20,21,22,23,24,25,26])" \
  116 +Turk(brokennose,[broken],3), \
  117 +Turk(beard,[nobeard,bigbeard,lightbeard,goatee,linebeard,normalbeard,lincolnbeard],3), \
  118 +Turk(chinsize,[shortchin,longchin],3)])" \
83 119 -path $DATA/CUFSF/target/ -train results1194v2.turk $ATTDIR/all.model
84 120 fi
85 121  
... ... @@ -90,7 +126,6 @@ br -crossValidate 2 -convert Output $ATTDIR/all.mtx $ATTDIR/all.rank
90 126 br -crossValidate 2 -convert Output $ATTDIR/all.mtx $ATTDIR/all_CUFSF%1.eval
91 127  
92 128 # Not trained on: earpitch, earsize, neck thickness
93   -# Not used for comparison: gender, eyelashvisbility
94 129  
95 130 #br -crossValidate 2 -path $DATA/CUFSF/target/ -algorithm models/attributes -enroll results30v2.turk gallery/results30v2.gal
96 131  
... ...