def has_2ch(f):
ch_slices = glob.glob(f+"/2ch_*.pkl")
if len(ch_slices) > 0:
patient_id = str(data_loader._extract_id_from_path(ch_slices[0]))
slice_name = os.path.basename(ch_slices[0])
heart_size = max(float(d[patient_id][slice_name]['roi_radii'][0]) / c[patient_id][slice_name]['PixelSpacing'][0],
float(d[patient_id][slice_name]['roi_radii'][1]) / c[patient_id][slice_name]['PixelSpacing'][1])
return (heart_size>=32)
else:
return False
def has_4ch(f):
ch_slices = glob.glob(f+"/4ch_*.pkl")
if len(ch_slices) > 0:
patient_id = str(data_loader._extract_id_from_path(ch_slices[0]))
slice_name = os.path.basename(ch_slices[0])
heart_size = max(float(d[patient_id][slice_name]['roi_radii'][0]) / c[patient_id][slice_name]['PixelSpacing'][0],
float(d[patient_id][slice_name]['roi_radii'][1]) / c[patient_id][slice_name]['PixelSpacing'][1])
return (heart_size<32)
else:
return False
def has_4ch(f):
ch_slices = glob.glob(f + "/4ch_*.pkl")
if len(ch_slices) > 0:
patient_id = str(data_loader._extract_id_from_path(ch_slices[0]))
slice_name = os.path.basename(ch_slices[0])
heart_size = max(
float(d[patient_id][slice_name]["roi_radii"][0]) / c[patient_id][slice_name]["PixelSpacing"][0],
float(d[patient_id][slice_name]["roi_radii"][1]) / c[patient_id][slice_name]["PixelSpacing"][1],
)
return heart_size < 32
else:
return False
