#!/usr/bin/python
import time
import sys
import os
from os import walk
import random

import numpy as np

if len(sys.argv) != 2:
    print(
        "This script will make img_path --- > person_id  with random \n"
        "For example, for folder 1,  then run:\n "
        " ./xxx.py 1 \n")
    exit()

folder_number = sys.argv[1]
aa = {}
bb = []
person = {}
count = 0

oripath = "./identities_"+folder_number+"/"


for (dirpath, dirnames, filenames) in walk(oripath):
  for dir in dirnames:
     person[dir] = count
     for (dirpath1, dirnames1, filenames1) in walk(oripath+dir):
        for i in filenames1:
            aa.update({dirpath1+"/"+i : count})
     count = count + 1

bb =  aa.items()
random.shuffle(bb)
for k,v in bb:
   print k,"\t\t",v

aa.clear()
person.clear()

 

相关文章:

  • 2022-12-23
  • 2021-08-31
  • 2021-12-23
  • 2021-05-01
  • 2022-01-14
  • 2022-12-23
  • 2021-06-22
  • 2021-08-25
猜你喜欢
  • 2021-09-12
  • 2021-06-08
  • 2021-06-15
  • 2022-12-23
  • 2021-07-02
  • 2021-11-15
  • 2021-07-25
相关资源
相似解决方案