mirror of
https://github.com/jlengrand/Ivolution.git
synced 2026-03-10 08:21:18 +00:00
eac2a7727f
Some problems with video creation. Cannot be read on vlc and wmp. TODO : Create executables Allow saving sessions (will save lots of my time) Ant script, test class (will also save some time!)
201 lines
7.1 KiB
Python
201 lines
7.1 KiB
Python
'''
|
|
Created on 27 mars 2012
|
|
|
|
@author: jll
|
|
'''
|
|
import cv
|
|
import os
|
|
import Guy
|
|
from FaceParams import FaceParams
|
|
|
|
class FaceMovie(object):
|
|
'''
|
|
Main class of the whole application.
|
|
Contains the core image processing functions.
|
|
Supports the communication layer with the end user interface.
|
|
'''
|
|
def __init__(self, in_folder, out_folder, param_folder):
|
|
'''
|
|
Constructor
|
|
'''
|
|
self.source= in_folder # Source folder for pictures
|
|
self.out = out_folder # Folder to save outputs
|
|
self.params_source = param_folder # Folder in which xml files can be found
|
|
|
|
self.guys = [] # List of pictures in source folder
|
|
|
|
# Setting up some default parameters for Face Detection
|
|
self.face_params = FaceParams(self.params_source)
|
|
|
|
# Position of the center in output images
|
|
self.x_center = 0
|
|
self.y_center = 0
|
|
|
|
# minimum size needed on right of center
|
|
self.x_af = 0
|
|
self.y_af = 0
|
|
|
|
# Needed minimum size of output image
|
|
self.dim_x = 0
|
|
self.dim_y = 0
|
|
|
|
def list_guys(self):
|
|
"""
|
|
Aims at populating the guys list, using the source folder as an input.
|
|
Guys list shall be sorted by file name alphabetical order
|
|
"""
|
|
try:
|
|
os.path.exists(self.source)
|
|
os.path.isdir(self.source) # checking if folder exists
|
|
except : # find precise exception
|
|
print "ERROR : Source folder not found ! Exiting. . ."
|
|
sys.exit(0)
|
|
|
|
# just listing directory. Lets be more secure later
|
|
files = os.listdir(self.source)
|
|
|
|
# loading images, create Guys and store it into guys
|
|
for token in files :
|
|
image = cv.LoadImage(os.path.join(self.source, token))
|
|
guy_name = os.path.splitext(token)[0]
|
|
a_guy = Guy.Guy(image, guy_name)
|
|
|
|
# populating guys
|
|
self.guys.append(a_guy)
|
|
|
|
def search_faces(self):
|
|
"""
|
|
Searches for all faces in the guys we have
|
|
Results to be stored directly in guys
|
|
"""
|
|
for a_guy in self.guys:
|
|
a_guy.search_face(self.face_params)
|
|
if a_guy.has_face(): # face(s) have been found
|
|
print "%d faces found for %s" % (a_guy.num_faces(), a_guy.name)
|
|
else:
|
|
print "Warning! No face found for %s" %(a_guy.name)
|
|
|
|
def normalize_faces(self, reference=0):
|
|
"""
|
|
Creates new images, normalized by face size
|
|
"""
|
|
if reference == 0:
|
|
reference = self.guys[0].faces[0][0][3] # catch face size (width)
|
|
|
|
for a_guy in self.guys:
|
|
if a_guy.has_face():
|
|
a_guy.normalize_face(reference)
|
|
|
|
def find_out_dims(self):
|
|
"""
|
|
Calculates best output image size and position depending on
|
|
faces found in guys.
|
|
"""
|
|
# FIXME: badly done !
|
|
for a_guy in self.guys:
|
|
if a_guy.has_face():
|
|
if a_guy.normalize:
|
|
xc = a_guy.x_norm_center
|
|
yc = a_guy.y_norm_center
|
|
inx = a_guy.norm_x
|
|
iny = a_guy.norm_y
|
|
else:
|
|
xc = a_guy.x_center
|
|
yc = a_guy.y_center
|
|
inx = a_guy.in_x
|
|
iny = a_guy.in_y
|
|
|
|
# update center
|
|
if xc > self.x_center:
|
|
self.x_center = xc
|
|
if yc > self.y_center:
|
|
self.y_center = yc
|
|
# update right part
|
|
if (inx - xc) > self.x_af:
|
|
self.x_af = inx - xc
|
|
if (iny - yc) > self.y_af:
|
|
self.y_af = iny - yc
|
|
|
|
self.dim_x = self.x_af + self.x_center
|
|
self.dim_y = self.y_af + self.y_center
|
|
|
|
def show_faces(self, time=1000, equalize=True):
|
|
"""
|
|
Show all faces that have been found for the guys.
|
|
The time for which each image will be displayed can be chosen.
|
|
Several modes can be chosen to adapt the result.
|
|
"""
|
|
for a_guy in self.guys:
|
|
if a_guy.has_face():
|
|
a_guy.create_video_output(self.dim_x,
|
|
self.dim_y,
|
|
self.x_center,
|
|
self.y_center)
|
|
a_guy.out_display(time)
|
|
|
|
def save_faces(self, out_folder, im_format="png"):
|
|
"""
|
|
Save all faces into out_folder, in the given image format
|
|
Debug is used to draw rectangles around found faces
|
|
"""
|
|
for a_guy in self.guys:
|
|
if a_guy.has_face():
|
|
a_guy.create_video_output(self.dim_x,
|
|
self.dim_y,
|
|
self.x_center,
|
|
self.y_center)
|
|
a_guy.save_result(out_folder, im_format)
|
|
|
|
def save_movie(self, out_folder, equalize=True):
|
|
"""
|
|
Creates a movie with all faces found in the inputs.
|
|
Guy is skipped if no face is found.
|
|
|
|
TODO : No codec involved !
|
|
TODO : Can FPS be changed?
|
|
Resize should be done somewhere else !
|
|
"""
|
|
filename = os.path.join(out_folder, "output.avi")
|
|
fourcc = 0#-1
|
|
fps = 24 # not taken into account
|
|
frameSize = (self.dim_x, self.dim_y)
|
|
my_video = cv.CreateVideoWriter(filename,
|
|
fourcc,
|
|
fps,
|
|
frameSize,
|
|
1)
|
|
for a_guy in self.guys:
|
|
if a_guy.has_face():
|
|
a_guy.create_video_output(self.dim_x,
|
|
self.dim_y,
|
|
self.x_center,
|
|
self.y_center)
|
|
cv.WriteFrame(my_video, a_guy.out_im)
|
|
|
|
def number_guys(self):
|
|
"""
|
|
Simply returns the number of guys in the current to-be movie
|
|
"""
|
|
return len(self.guys)
|
|
|
|
if __name__ == "__main__":
|
|
# quick and dirty tests
|
|
root_fo = "C:\Users\jll\perso\workspace\FaceMovie"
|
|
#in_fo = os.path.join(root_fo, "input\Axel_tsts")
|
|
in_fo = os.path.join(root_fo, "input\Axel")
|
|
out_fo = os.path.join(root_fo, "output")
|
|
par_fo = os.path.join(root_fo, "haarcascades")
|
|
|
|
my_movie = FaceMovie(in_fo, out_fo, par_fo)
|
|
my_movie.list_guys() # find images in input folder
|
|
my_movie.search_faces() # search for images with faces
|
|
# I want to change images so that all faces have the same size
|
|
my_movie.normalize_faces() # sets all faces to the same size
|
|
# I want to know the size of the output frame, knowing initial conditions
|
|
my_movie.find_out_dims() # finds output minimal size to get all eyes in the same place
|
|
|
|
#my_movie.show_faces(1000)
|
|
#my_movie.save_faces("output")
|
|
#my_movie.save_movie("output")
|
|
|
|
print "Facemovie finished !" |