MS-Celeb-1M 数据

MS-Celeb-1M数据集

解压数据集：还未测试

      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))
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/115135028