You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.8 KiB
48 lines
1.8 KiB
2 years ago
|
import os
|
||
|
|
from tqdm import tqdm
|
||
|
|
|
||
|
|
os.system('rm -rf images')
|
||
|
|
os.system('mkdir images')
|
||
|
|
|
||
|
|
os.system('cp -r ../MME_Benchmark_release/OCR images/')
|
||
|
|
|
||
|
|
os.system('mkdir images/artwork')
|
||
|
|
os.system('cp ../MME_Benchmark_release/artwork/questions_answers_YN/* images/artwork/')
|
||
|
|
with open('LaVIN/artwork.txt') as fin:
|
||
|
|
paths = [ line.strip().split('\t', 1)[0] for line in fin ]
|
||
|
|
paths = list(set(paths))
|
||
|
|
for path in tqdm(paths):
|
||
|
|
os.system(f'cp ../MME_Benchmark_release/artwork/images/toy_dataset/{path} images/artwork/{path}')
|
||
|
|
|
||
|
|
os.system('mkdir images/celebrity')
|
||
|
|
os.system('cp ../MME_Benchmark_release/celebrity/images/* images/celebrity/')
|
||
|
|
os.system('cp ../MME_Benchmark_release/celebrity/questions_answers_YN/* images/celebrity/')
|
||
|
|
|
||
|
|
os.system('cp -r ../MME_Benchmark_release/code_reasoning images/')
|
||
|
|
|
||
|
|
os.system('cp -r ../MME_Benchmark_release/color images/')
|
||
|
|
|
||
|
|
os.system('cp -r ../MME_Benchmark_release/commonsense_reasoning images/')
|
||
|
|
|
||
|
|
os.system('cp -r ../MME_Benchmark_release/count images/')
|
||
|
|
|
||
|
|
os.system('cp -r ../MME_Benchmark_release/existence images/')
|
||
|
|
|
||
|
|
os.system('mkdir images/landmark')
|
||
|
|
os.system('cp ../MME_Benchmark_release/landmark/images/* images/landmark/')
|
||
|
|
os.system('cp ../MME_Benchmark_release/landmark/questions_answers_YN/* images/landmark/')
|
||
|
|
|
||
|
|
os.system('cp -r ../MME_Benchmark_release/numerical_calculation images/')
|
||
|
|
|
||
|
|
os.system('cp -r ../MME_Benchmark_release/position images/')
|
||
|
|
|
||
|
|
os.system('mkdir images/posters')
|
||
|
|
os.system('cp ../MME_Benchmark_release/posters/images/* images/posters/')
|
||
|
|
os.system('cp ../MME_Benchmark_release/posters/questions_answers_YN/* images/posters/')
|
||
|
|
|
||
|
|
os.system('mkdir images/scene')
|
||
|
|
os.system('cp ../MME_Benchmark_release/scene/images/* images/scene/')
|
||
|
|
os.system('cp ../MME_Benchmark_release/scene/questions_answers_YN/* images/scene/')
|
||
|
|
|
||
|
|
os.system('cp -r ../MME_Benchmark_release/text_translation images/')
|