24 lines
650 B
Python
24 lines
650 B
Python
|
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()
|