Skip to content

Latest commit

 

History

History
12 lines (7 loc) · 996 Bytes

ChangeDataset.md

File metadata and controls

12 lines (7 loc) · 996 Bytes
Raw

How to run the code on other datasets?

  1. Based on the production process of the IXI dataset in TransMorph or the LPBA dataset we provided, create your corresponding .pkl dataset. Or modify the file reading method in data/datasets.py according to your needs.

  2. Modify the self.seg_table of the class Seg_norm in data/trans.py to include all the values present in the label images of the dataset, which can typically be obtained using the np.unique function.

    image

    The function of this class is to normalize the values in seg_table to a sequence of consecutive integers starting from zero.

  3. Modify the VOI_lbls in the dice_val_VOI function of the utils.py to the number of the label (normalized in step 2) that needs to be calculated in Dice metric.

    image