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