利用Python处理ReId的公开数据集,把quary数据拷贝到每个单独的文件夹中,地库数据也拷贝到单独的文件夹中,一个文件夹一个reid。
#coding=utf-8
import cv2
import os
import shutil
lib_images_src = "./query"
test_images_src = "./bounding_box_test"
lib_images_dest = "./test_images"
test_images_dest = "./lib_images"
def mkSaveDir(image_root, extra=''):
#save_dir = image_root + "_save"
save_dir = image_root + extra
print("save_dir in mkSaveDir function:", save_dir)
if not os.path.exists(save_dir):
os.mkdir(save_dir)
return save_dir
def mkChildDir(image_root, child=''):
#save_dir = image_root + "_save"
save_dir = os.path.join(image_root, child)
print("save_dir in mkSaveDir function:", save_dir)
if not os.path.exists(save_dir):
os.mkdir(save_dir)
return save_dir
def copyImag