Peter M. Groen / openbr

Commit f722baa332cf229461fc653e7b0e85126ec3b58b

Authored by Josh Klontz
1 parent f2b5f8a3

simplified algorithm by introducing TurkClassifierTransform

Inline Side-by-side
Showing 2 changed files with 88 additions and 49 deletions
openbr/plugins/turk.cpp
  View file @f722baa
@@ -57,6 +57,58 @@ BR_REGISTER(Transform, TurkTransform) @@ -57,6 +57,58 @@ BR_REGISTER(Transform, TurkTransform)
57 57
58 /*! 58 /*!
59 * \ingroup transforms 59 * \ingroup transforms
  60 + * \brief Convenience class for training turk attribute regressors
  61 + * \author Josh Klontz \cite jklontz
  62 + */
  63 +class TurkClassifierTransform : public Transform
  64 +{
  65 + Q_OBJECT
  66 + Q_PROPERTY(QString key READ get_key WRITE set_key RESET reset_key STORED false)
  67 + Q_PROPERTY(QStringList values READ get_values WRITE set_values RESET reset_values STORED false)
  68 + Q_PROPERTY(float maxVotes READ get_maxVotes WRITE set_maxVotes RESET reset_maxVotes STORED false)
  69 + BR_PROPERTY(QString, key, QString())
  70 + BR_PROPERTY(QStringList, values, QStringList())
  71 + BR_PROPERTY(float, maxVotes, 1)
  72 +
  73 + Transform *child;
  74 +
  75 + void init()
  76 + {
  77 + QString algorithm = QString("Turk(%1, %2)+").arg(key, QString::number(maxVotes));
  78 + QStringList classifiers;
  79 + foreach (const QString &value, values)
  80 + classifiers.append(QString("SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=%1,outputVariable=predicted_%1)").arg(value));
  81 + algorithm += classifiers.join("/");
  82 + if (values.size() > 1)
  83 + algorithm += "+Cat";
  84 + child = Transform::make(algorithm);
  85 + }
  86 +
  87 + void train(const QList<TemplateList> &data)
  88 + {
  89 + child->train(data);
  90 + }
  91 +
  92 + void project(const Template &src, Template &dst) const
  93 + {
  94 + child->project(src, dst);
  95 + }
  96 +
  97 + void store(QDataStream &stream) const
  98 + {
  99 + child->store(stream);
  100 + }
  101 +
  102 + void load(QDataStream &stream)
  103 + {
  104 + child->load(stream);
  105 + }
  106 +};
  107 +
  108 +BR_REGISTER(Transform, TurkClassifierTransform)
  109 +
  110 +/*!
  111 + * \ingroup transforms
60 * \brief Converts metadata into a map structure 112 * \brief Converts metadata into a map structure
61 * \author Scott Klum \cite sklum 113 * \author Scott Klum \cite sklum
62 */ 114 */
scripts/attributes.sh
  View file @f722baa
@@ -18,55 +18,42 @@ mkdir -p $ATTDIR @@ -18,55 +18,42 @@ mkdir -p $ATTDIR
18 DATA=${DATA:-~/data/CUHK-VHDC} 18 DATA=${DATA:-~/data/CUHK-VHDC}
19 19
20 if [ ! -f $ATTDIR/all.model ]; then 20 if [ ! -f $ATTDIR/all.model ]; then
21 - br -gui -crossValidate 2 -algorithm "CrossValidate($BASE+ \  
22 -($BROW+ \  
23 -(Turk(unibrow,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=unibrow,outputVariable=predicted_unibrow)+Cat)/ \  
24 -(Turk(eyebroworientation,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=eyebrowsdown,outputVariable=predicted_eyebrowsdown)/ \  
25 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=eyebrowsuptodown,outputVariable=predicted_eyebrowsuptodown)+Cat)/ \  
26 -(Turk(thickeyebrows,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=thickeyebrows,outputVariable=predicted_thickeyebrows)/ \  
27 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=lighteyebrows,outputVariable=predicted_lighteyebrows)+Cat))/ \  
28 -($MOUTH+ \  
29 -(Turk(smiling,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=smiling,outputVariable=predicted_smiling)+Cat)/ \  
30 -(Turk(mouthasymmetry,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=asymmetrical,outputVariable=predicted_asymmetrical)+Cat))/ \  
31 -($EYES+ \  
32 -(Turk(eyecolor,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=darkeyes,outputVariable=predicted_darkeyes)/ \  
33 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=lighteyes,outputVariable=predicted_lighteyes)+Cat)/ \  
34 -(Turk(baggyeyes,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=baggy,outputVariable=predicted_baggy)+Cat)/ \  
35 -(Turk(almondeyes,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=almond,outputVariable=predicted_almond)+Cat)/ \  
36 -(Turk(buriedeyes,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=buriedeyes,outputVariable=predicted_buriedeyes)+Cat)/ \  
37 -(Turk(sleepyeyes,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=sleepy,outputVariable=predicted_sleepy)+Cat)/ \  
38 -(Turk(lineeyes,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=line,outputVariable=predicted_line)+Cat)/ \  
39 -(Turk(roundeyes,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=round,outputVariable=predicted_round)+Cat)/ \  
40 -(Turk(smalleyes,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=smalleyes,outputVariable=predicted_smalleyes)+Cat)/ \  
41 -(Turk(glasses,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=glasses,outputVariable=predicted_glasses)+Cat)/ \  
42 -(Turk(eyelashvisibility,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=feweyelashes,outputVariable=predicted_feweyelashes)+Cat))/ \  
43 -($FACE+ \  
44 -(Turk(gender,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=male,outputVariable=predicted_male)+Cat)/ \  
45 -(Turk(cheekdensity,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=puffy,outputVariable=predicted_puffy)/ \  
46 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=in,outputVariable=predicted_in)/ \  
47 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=normal,outputVariable=predicted_normal)+Cat)/ \  
48 -(Turk(facemarks,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=scars,outputVariable=predicted_scars)/ \  
49 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=moles,outputVariable=predicted_moles)/ \  
50 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=normal,outputVariable=predicted_normal)+Cat)/ \  
51 -(Turk(facelength,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=long,outputVariable=predicted_long)+Cat)/ \  
52 -(Turk(nosetomouthdist,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=long,outputVariable=predicted_long)/ \  
53 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=small,outputVariable=predicted_small)+Cat))/ \  
54 -($HAIR+ \  
55 -(Turk(foreheadwrinkles,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=wrinkled,outputVariable=predicted_wrinkled)+Cat)/ \  
56 -(Turk(haircolor,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=darkhair,outputVariable=predicted_darkhair)/ \  
57 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=lighthair,outputVariable=predicted_lighthair)/ \  
58 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=greyhair,outputVariable=predicted_greyhair)+Cat)/ \  
59 -(Turk(hairstyle,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=curlyhair,outputVariable=predicted_curlyhair)+Cat))/ \  
60 -($NOSE+ \  
61 -(Turk(noseorientation,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=upnose,outputVariable=predicted_upnose)/ \  
62 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=downnose,outputVariable=predicted_downnose)+Cat)/ \  
63 -(Turk(nosewidth,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=small,outputVariable=predicted_small)/ \  
64 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=thick,outputVariable=predicted_thick)+Cat)/ \  
65 -(Turk(nosesize,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=smallnose,outputVariable=predicted_smallnose)/ \  
66 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=bignose,outputVariable=predicted_bignose)+Cat))/ \  
67 -($JAW+ \  
68 -(Turk(chinsize,3)+SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=shortchin,outputVariable=predicted_shortchin)/ \  
69 -SVM(RBF,EPS_SVR,returnDFVal=true,inputVariable=longchin,outputVariable=predicted_longchin))+Cat)): \ 21 + br -crossValidate 2 -algorithm "CrossValidate($BASE+ \
  22 + ($BROW+ \
  23 + TurkClassifier(unibrow,[unibrow],3)/ \
  24 + TurkClassifier(eyebroworientation,[eyebrowsdown,eyebrowsuptodown],3)/ \
  25 + TurkClassifier(thickeyebrows,[thickeyebrows,lighteyebrows],3))/ \
  26 + ($MOUTH+ \
  27 + TurkClassifier(smiling,[smiling],3)/ \
  28 + TurkClassifier(mouthasymmetry,[asymmetrical],3))/ \
  29 + ($EYES+ \
  30 + TurkClassifier(eyecolor,[darkeyes,lighteyes],3)/ \
  31 + TurkClassifier(baggyeyes,[baggy],3)/ \
  32 + TurkClassifier(almondeyes,[almond],3)/ \
  33 + TurkClassifier(buriedeyes,[buriedeyes],3)/ \
  34 + TurkClassifier(sleepyeyes,[sleepy],3)/ \
  35 + TurkClassifier(lineeyes,[line],3)/ \
  36 + TurkClassifier(roundeyes,[round],3)/ \
  37 + TurkClassifier(smalleyes,[smalleyes],3)/ \
  38 + TurkClassifier(glasses,[glasses],3)/ \
  39 + TurkClassifier(eyelashvisibility,[feweyelashes],3))/ \
  40 + ($FACE+ \
  41 + TurkClassifier(gender,[male],3)/ \
  42 + TurkClassifier(cheekdensity,[puffy,in,normal],3)/ \
  43 + TurkClassifier(facemarks,[scars,moles,normal],3)/ \
  44 + TurkClassifier(facelength,[long],3)/ \
  45 + TurkClassifier(nosetomouthdist,[long,small],3))/ \
  46 + ($HAIR+ \
  47 + TurkClassifier(foreheadwrinkles,[wrinkled],3)/ \
  48 + TurkClassifier(haircolor,[darkhair,lighthair,greyhair],3)/ \
  49 + TurkClassifier(hairstyle,[curlyhair],3))/ \
  50 + ($NOSE+ \
  51 + TurkClassifier(noseorientation,[upnose,downnose],3)/ \
  52 + TurkClassifier(nosewidth,[small,thick],3)/ \
  53 + TurkClassifier(nosesize,[smallnose,bignose],3))/ \
  54 + ($JAW+ \
  55 + TurkClassifier(chinsize,[shortchin,longchin],3)) \
  56 + ): \
70 CrossValidate+Fuse([ \ 57 CrossValidate+Fuse([ \
71 Turk(unibrow,[unibrow],3), \ 58 Turk(unibrow,[unibrow],3), \
72 Turk(eyebroworientation,[eyebrowsdown,eyebrowsuptodown],3), \ 59 Turk(eyebroworientation,[eyebrowsdown,eyebrowsuptodown],3), \