re_index.py

# -*- coding: utf-8 -*-
# change name of the folder(e.g.  0002,0007,0010,0011...  to 0,1,2,3)
import numpy as np
import os
from shutil import copyfile

original_path='/home/paul/datasets/duke/pytorch'

#copy folder tree from source to destination
def copyfolder(src,dst):
    files=os.listdir(src)
    if not os.path.isdir(dst):
        os.mkdir(dst)
    for tt in files:   
        copyfile(src+'/'+tt,dst+'/'+tt)

train_save_path = original_path + '/val_new'
if not os.path.exists(train_save_path):
    os.mkdir(train_save_path)

data_path=original_path+'/val'
if not os.path.isdir(train_save_path):
    os.mkdir(train_save_path)

reid_index=0
folders=os.listdir(data_path)
folders = sorted(folders)

for foldernames in folders:
    copyfolder(data_path+'/'+foldernames,train_save_path+'/'+str(reid_index).zfill(4))
    reid_index=reid_index+1