將wiki人臉數據集中的圖片按男女分類

import shutil
f_in = 'D:/wiki_crop'
# 讀取文件中圖片信息根據性別分類圖片到對應目錄中
dirroot = "D:/"
f = open(dirroot+"\\lable2.txt","r")
s = 0
i=1
print(f)
# line = f.readlines()
# print(line[1])

for line in f.readlines():
    line = line.strip().split(' ')

    s = s + 1
    # print(type(line))
    # print(line[3])
    # dir = list(line)

    # imgName =  line[i]+line[i+1]

    imgName = line[0]
    # print(line[0])
    print(s)
    print(f_in+'/'+imgName)
    num= str(s)
    if i > 0:
        if line[3]== "0.0":
            print("female")
            try:
                shutil.copy(f_in+'/'+imgName, "D:/data/female/"+ num+ '.jpg')
            except:
                pass
        #     移動圖片到female目錄
        elif line[3]=="1.0":
            print("male")
            try:
                shutil.copy(f_in+'/'+imgName, "D:/data/male/" + num+ '.jpg')
            except:
                pass
        #     移動圖片到male目錄
        else:
            print("N")
            #未識別男女

f.close()

附上自己從wiki.mat中提取出來的標簽

https://pan.baidu.com/s/1dGT6ghV

再附上wiki的數據集地址

https://data.vision.ee.ethz.ch/cvl/rrothe/imdb-wiki/