jlengrand/Ivolution
1
0
Fork 0
mirror of https://github.com/jlengrand/Ivolution.git synced 2026-03-10 08:21:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Changes training types to get one word keys. Adds available training files listing option

Browse Source
This commit is contained in:
AirballTest
2012-05-16 13:56:07 +02:00
parent 4f47cb895e
commit 8d417607ee
2 changed files with 26 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
facemovie/Facemoviefier.py
View File

@@ -6,6 +6,7 @@ Created on 29 mars 2012
# This file should never be imported anywhere
import os
import sys
import argparse
import Facemovie
@@ -22,10 +23,10 @@ class Facemoviefier():
self.args = self.initCLParser()
print self.args
# par folder should be known (contained somewhere in the installation)
par_fo = os.path.join(self.args['root'], "haarcascades")
def init_facemovie(self):
# FIXME : par folder should be known (contained somewhere in the installation)
par_fo = os.path.join(self.args['root'], "haarcascades")
self.face_params = FaceParams.FaceParams(par_fo, self.args['param'])
self.facemovie = Facemovie.FaceMovie(self.args['input'], self.args['output'], self.face_params)
def initCLParser(self):
@@ -78,20 +79,34 @@ class Facemoviefier():
default='n')
# TODO: Integrate face params file choice, with list of possibilities.
params = training_types.simple_set.keys()
params.append('?')
parser.add_argument('-p',
'--param',
choices=training_types.simple_set,
choices=params,
help='Choose the desired file for training the recognition phaze. Should be chosen depending on the face presentation (profile, whole body, ...)',
default='frontal face alt')
default='frontal_facee')
args = vars(parser.parse_args())
return args
def run(self):
"""
Runs all the steps needed to get the desired output
"""
# selects xml used to train the classifier
if self.args['param'] == '?':
print "Available param files are :"
print "==="
for item in training_types.simple_set:
print item
print "==="
print 'Please choose your param file (or let default value) and restart the application'
sys.exit(0)
else:
# creates Facemovie object, loads param file
self.init_facemovie()
#selects sorting method
if self.args['sort'] == 'e':
self.facemovie.sort_method = 'e';

10
facemovie/training_types.py
View File

@@ -34,11 +34,11 @@ complete_set = {#eyes
}
simple_set = {# frontal faces
'frontal face alt':"haarcascade_frontalface_alt",
'frontal_face':"haarcascade_frontalface_alt",
#profile face
'profile face':"haarcascade_profileface",
'profile_face':"haarcascade_profileface",
#body
'full body':"haarcascade_fullbody",
'lower body':"haarcascade_lowerbody",
'upper body':"haarcascade_upperbody",
'full_body':"haarcascade_fullbody",
'lower_body':"haarcascade_lowerbody",
'upper_body':"haarcascade_upperbody",
}