import shutil
import glob
import os
import json
path = r'/root/projects/TransT/data/Lasot'
dest = '/root/projects/TransT/data/Lasot/test'
path1 = os.path.join(path, "LaSOTBenchmark")
file_lst = glob.glob(path1 + '/*')
f = open('/root/projects/TransT/data/Lasot/test/LaSOT.json','r')
data = json.load(f)
f.close()
def copy_files():
for file in file_final11_lst:
# print(file)
filename = file.split('/')[-1]
# print(filename)
# filename = os.path.splitext(file)[0]
if filename in data:
print("copy:" , filename)
os.mkdir(os.path.join(dest, filename))
# print(file)
src_files = os.listdir(file)
for file_name in src_files:
full_file_name = os.path.join(file, file_name)
if os.path.isfile(full_file_name):
shutil.copy(full_file_name, os.path.join(dest, filename))
else:
os.mkdir(os.path.join(dest, filename, file_name))
s_images = os.listdir(os.path.join(file, file_name))
for x in s_images:
shutil.copy(os.path.join(full_file_name, x), os.path.join(dest,filename,'img'))
for file_st in file_lst:
file_final_lst = glob.glob(file_st+'/')
# print(file_final_lst)
for file_final11 in file_final_lst:
# print(file_final11)
file_final11_lst = glob.glob(file_final11 + '*')
# print(file_final11_lst)
copy_files()