import os

from imaging.preprocessing import process_dataset

data_root   = '/mnt/1220/Public/dataset/Spine/CTSpine1K/data/'
label_root  = '/mnt/1220/Public/dataset/Spine/CTSpine1K/label/'
output_dir = '/mnt/1248/open2/cyrou/CBT/Seg/Resample/standardized-xfr/'

label_map = {
    'colon': 'conlon',
    'COVID-19': 'COVID-19',
    'HNSCC-3DCT-RT_neck': 'HNSCC-3DCT-RT_neck',
    'liver': 'Liver',
}

def main(): 

    for key, value in label_map.items():
        data_dir = os.path.join(data_root, key)
        label_dir = os.path.join(label_root, value)
        process_dataset(data_dir, label_dir, output_dir)

if __name__ == '__main__':
    main()