import glob
import os
import shutil
path = '/root/projects/TransT/data/Lasot/test'
data = ['dog', 'goldfish', 'deer', 'crab', 'coin', 'flag', 'crocodile', 'giraffe', 'fox', 'chameleon', 'gecko']
file_lst = glob.glob(path + '/*')
for file in file_lst:
# print(file)
filename = file.split('/')[-1]
# print(filename)
filename1 = filename.split('-')[0]
if filename1 in data:
print(file)
shutil.rmtree(file)