性别、年龄预测

首先按照链接：http://www.openu.ac.il/home/hassner/projects/cnn_agegender/下载所需的cnn_age_gender_models_and_data.0.0.2.zip到models。

先要保证python版本2.7以上http://yijiebuyi.com/blog/108ae6186bb00cc708bc54f02adec277.html

升级python后，可能导致pip不可用，执行以下命令即可

wget https://bootstrap.pypa.io/get-pip.py

python get-pip.py

numpy路径：/usr/lib64/python2.7/site-packages/numpy/core/include/numpy/arrayobject.h

之后需要编译caffe，使其支持python，具体编译方法见：http://caffe.berkeleyvision.org/installation.html。

预测代码

import os
import numpy as np
import shutil

caffe_root = '/opt/caffe/' 
import sys
sys.path.insert(0, caffe_root + 'python')
import caffe

def predict(src_folder):
    mean_filename='/opt/caffe/models/AgeGenderCNN/mean.binaryproto'
    proto_data = open(mean_filename, "rb").read()
    a = caffe.io.caffe_pb2.BlobProto.FromString(proto_data)
    mean  = caffe.io.blobproto_to_array(a)[0]

    age_net_pretrained='/opt/caffe/models/AgeGenderCNN/age_net.caffemodel'
    age_net_model_file='/opt/caffe/models/AgeGenderCNN/deploy_age.prototxt'
    age_net = caffe.Classifier(age_net_model_file, age_net_pretrained,
                           mean=mean,
                           channel_swap=(2,1,0),
                           raw_scale=255,
                           image_dims=(256, 256))

    gender_net_pretrained='/opt/caffe/models/AgeGenderCNN/gender_net.caffemodel'
    gender_net_model_file='/opt/caffe/models/AgeGenderCNN/deploy_gender.prototxt'
    gender_net = caffe.Classifier(gender_net_model_file, gender_net_pretrained,
                           mean=mean,
                           channel_swap=(2,1,0),
                           raw_scale=255,
                           image_dims=(256, 256))

    age_list=['(0, 2)','(4, 6)','(8, 12)','(15, 20)','(25, 32)','(38, 43)','(48, 53)','(60, 100)']
    gender_list=['Male','Female']
    gender_folder = 'male'
    for people_folder in os.listdir(src_folder):
        people_path = src_folder + people_folder + '/'
        for img_file in os.listdir(people_path):
            img_path = people_path+img_file
            input_image = caffe.io.load_image(img_path)
#            prediction = age_net.predict([input_image]) 
#            print 'predicted age:', age_list[prediction[0].argmax()]
            prediction = gender_net.predict([input_image])
#            print 'predicted gender:', gender_list[prediction[0].argmax()]
            if gender_list[prediction[0].argmax()] != gender_folder:
                print 'processing img:',img_path ,'gender:',gender_list[prediction[0].argmax()],' prediction:',prediction
                if gender_folder == 'Male':
                    shutil.copy(img_path,src_folder+'../maleout')
                elif gender_folder == 'Female':
                    shutil.copy(img_path,src_folder+'../femaleout')

if __name__ == '__main__':
    if len(sys.argv) != 2:
        print 'Usage: python %s src_folder' % (sys.argv[0])
        sys.exit()
    src_folder = sys.argv[1]
    if not src_folder.endswith('/'):
        src_folder += '/'
    predict(src_folder)