import os
import shutil

# SPLIT = '613'
# SPLIT = 'genre'
SPLIT = 'cat'

OUTPUT_PATH = './output'
RESULT_PATH = f'../../evaluation/results/baselines/{SPLIT}/'
TASKS = ['interaction', 'semantics', 'interactable']
IS_FINETUNE = ['', '_finetune']
FORMATS = ['det']

for task in TASKS:
    for format in FORMATS:
        for is_finetune in IS_FINETUNE:
            result_path = os.path.join(RESULT_PATH, format, task)
            os.makedirs(result_path, exist_ok=True)
            output_path = os.path.join(OUTPUT_PATH, format, task + is_finetune, SPLIT)
            if not os.path.exists(output_path):
                continue
            shutil.copy2(os.path.join(output_path, 'inference_test/coco_instances_results.json'), 
                         os.path.join(result_path, 'CenterNet2' + is_finetune + '.json'))