The emotion-style from botcs

emotion-style's People

Contributors

Watchers

emotion-style's Issues

imdb_crop gender split error

Hi, I used your data_loader.py to get the full_path and gender information. I want to split the imdb_crop into female and male folder. But it seems to be wrong. The code lists as follows:

def load_imdbmat():
    dataset = loadmat('./imdb_crop/imdb.mat')
    fid=open('imdb.txt','w')
    face_score_threshold=3
    image_names_array = dataset['imdb']['full_path'][0, 0][0]
    gender_classes = dataset['imdb']['gender'][0, 0][0]
    face_score = dataset['imdb']['face_score'][0, 0][0]
    second_face_score = dataset['imdb']['second_face_score'][0, 0][0]
    face_score_mask = face_score > face_score_threshold
    second_face_score_mask = np.isnan(second_face_score)
    unknown_gender_mask = np.logical_not(np.isnan(gender_classes))
    mask = np.logical_and(face_score_mask, second_face_score_mask)
    mask = np.logical_and(mask, unknown_gender_mask)
    image_names_array = image_names_array[mask]
    gender_classes = gender_classes[mask].tolist()
    image_names = []
    for image_name_arg in range(image_names_array.shape[0]):
        image_name = image_names_array[image_name_arg][0]
        image_names.append(image_name)
    #print 'image_names:',len(image_names)
    #print 'gender:',len(gender_classes)
    for img_name,gender in zip(image_names,gender_classes):
        line=img_name+" "+str(int(gender))+"\n"
        fid.writelines(line)
    print 'done'

The code above can run, but get an error result. It can't split the gender. Any help will be appreciated.

Recommend Projects