import base64
import csv
import os
filename = "K:\celib\dataset\MsCelebV1-Faces-Aligned.part.04.tsv"
outputDir = "K:\celib"
with open(filename, 'r') as tsvF:
reader = csv.reader(tsvF, delimiter='\t')
i = 0
for row in reader:
MID, imgSearchRank, faceID, data = row[0], row[1], row[4], base64.b64decode(row[-1])
saveDir = os.path.join(outputDir, MID)
savePath = os.path.join(saveDir, "{}-{}.jpg".format(imgSearchRank, faceID))
if not os.path.exists(saveDir):
os.mkdir(saveDir)
with open(savePath, 'wb') as f:
f.write(data)
i += 1
if i % 1000 == 0:
print("Extracted {} images.".format(i))
解压下载的ms_celeb_1m数据集
最新推荐文章于 2024-05-30 23:41:26 发布