Omini3D / check_output.py
maxmo2009's picture
Sync from local: code + epoch-110 checkpoint, clean README
2af0e94 verified
import nibabel as nib
files = {
'128^3 original': 'Data/Reg_data/om/img/Patient0000_Slice000000_ORG_NA.nii.gz',
'128^3 registered': 'Data/Reg_data/om/img/Patient0000_Slice000000_AugImg0000_NoiseStep0075.nii.gz',
'Full-res original': 'Data/Reg_data/om/img_fullres/Patient0000_Slice000000_ORG_NA.nii.gz',
'Full-res registered':'Data/Reg_data/om/img_fullres/Patient0000_Slice000000_AugImg0000_NoiseStep0075.nii.gz',
'Full-res DDF': 'Data/Reg_data/om/ddf_fullres/Patient0000_Slice000000_AugImg0000_NoiseStep0075.nii.gz',
'Full-res mask orig': 'Data/Reg_data/om/msk_fullres/Patient0000_Slice000000_ORG_NA_GT.nii.gz',
'Full-res mask reg': 'Data/Reg_data/om/msk_fullres/Patient0000_Slice000000_AugImg0000_NoiseStep0075_GT.nii.gz',
}
for label, path in files.items():
img = nib.load(path)
print(f'{label:25s}: {img.shape}')