Permalink
Browse files

ms-celeb-1m: Use MID for directory, imgSearchRank+faceID for name.

  • Loading branch information...
1 parent 3582d54 commit 67e9c5d84f1ba1ae7af52b8c871295951a723695 @bamos bamos committed Jul 31, 2016
Showing with 3 additions and 3 deletions.
  1. +3 −3 data/ms-celeb-1m/extract.py
@@ -36,10 +36,10 @@
reader = csv.reader(tsvF, delimiter='\t')
i = 0
for row in reader:
- MID, faceID, data = row[0], row[4], base64.b64decode(row[-1])
+ MID, imgSearchRank, faceID, data = row[0], row[1], row[4], base64.b64decode(row[-1])
- saveDir = os.path.join(args.outputDir, faceID)
- savePath = os.path.join(saveDir, MID + '.jpg')
+ saveDir = os.path.join(args.outputDir, MID)
+ savePath = os.path.join(saveDir, "{}-{}.jpg".format(imgSearchRank, faceID))
# assert(magic.from_buffer(data) == 'JPEG image data, JFIF standard 1.01')

0 comments on commit 67e9c5d

Please sign in to comment.