You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
1.1 KiB
42 lines
1.1 KiB
"""
|
|
Programmer : EOF
|
|
File : mr_routine.py
|
|
Date : 2015.12.29
|
|
E-mail : jasonleaster@163.com
|
|
|
|
"""
|
|
from config import TRAINING_IMG_WIDTH
|
|
from config import TRAINING_IMG_HEIGHT
|
|
|
|
from haarFeature import Feature
|
|
import numpy
|
|
from functools import wraps
|
|
def processMeassure(fn):
|
|
@wraps(fn)
|
|
def measure_time(*args, **kwargs):
|
|
import os
|
|
import time
|
|
print "process ", os.getpid(), "started!"
|
|
start = time.time()
|
|
fn(*args, **kwargs)
|
|
end = time.time()
|
|
print "Cost time: ", end - start, " second."
|
|
print "Process " , os.getpid(), " end!"
|
|
|
|
return measure_time
|
|
|
|
@processMeassure
|
|
def routine(images, filename):
|
|
tot_samples = len(images)
|
|
|
|
haar = Feature(TRAINING_IMG_WIDTH, TRAINING_IMG_HEIGHT)
|
|
|
|
mat = numpy.zeros((haar.featuresNum, tot_samples), dtype = numpy.float32)
|
|
|
|
for i in range(tot_samples):
|
|
featureVec = haar.calFeatureForImg(images[i])
|
|
for j in range(haar.featuresNum):
|
|
mat[j][i] = featureVec[j]
|
|
|
|
numpy.save(filename, mat)
|