Exemple #1
0
 def handle_uploaded_file(self, udid, f, language):
     self.file = "{0}/{1}.jpg".format(self.image_upload_dir, udid)
     self.udid = udid
     self.language = language
     #save file on server
     with open(self.file, 'wb+') as destination:
         for chunk in f.chunks():
             destination.write(chunk)
     #save each part of face as separate file on server
     featuresExtractor = FeaturesExtractor()
     result_name = "{0}/{1}_result".format(self.image_upload_dir, udid)
     featuresExtractor.detect_and_draw(result_name, self.file, True)
    filename = name_arr[len(name_arr)-1]
    return filename[0:2]

#def get_ctr(input_name):
#    ctr = input_name[len(input_name)-5:len(input_name)-4]
#    return ctr


if __name__ == '__main__':
    
    if len(sys.argv) < 2:
        print "USAGE: features_extractor.py </path/to/input/images/> </path/to/output/folder/>"
        sys.exit()
        
    
    featuresExtractor = FeaturesExtractor()
    
    for file in glob.glob("{0}train/{1}*.jpg".format(settings.PROJECT_ROOT, sys.argv[2])):
        os.remove(file)
    #for input_name in glob.glob("{0}train/input/deleteme/*.jpg".format(settings.PROJECT_ROOT)):
    ctr = 0
    age = 0
    for input_name in glob.glob("{0}train/{1}*.jpg".format(settings.PROJECT_ROOT, sys.argv[1])):
        print input_name
        new_age = get_age(input_name)
        if age != new_age:
            ctr = 0
        age = new_age
        ctr = ctr + 1
        result_name = "{0}train/{1}{2}_{3}".format(settings.PROJECT_ROOT, sys.argv[2], age, ctr)
        featuresExtractor.detect_and_draw(result_name, input_name, False)