| ''' Example-1: manually list all dataset paths ''' | |
| # img_datas = [ | |
| # 'sam3d_train/medical_data_all/COVID_lesion/COVID1920_ct', | |
| # 'sam3d_train/medical_data_all/COVID_lesion/Chest_CT_Scans_with_COVID-19_ct', | |
| # 'sam3d_train/medical_data_all/adrenal/WORD_ct', | |
| # ] | |
| ''' Example-2: use glob to automatically list all dataset paths ''' | |
| from glob import glob | |
| import os.path as osp | |
| PROJ_DIR=osp.dirname(osp.dirname(__file__)) | |
| img_datas = glob(osp.join(PROJ_DIR, "data", "brain_pre_sam", "*", "*")) |