def view_mask(self):
data = SimpleDataInfo({'Original mask': load_nifti(self.selectedMaskText.text()).get_data(),
'Slice mask': load_nifti(self.selectedOutputFileText.text()).get_data()})
config = MapPlotConfig()
config.dimension = self.dimensionInput.value()
config.slice_index = self.sliceInput.value()
config.maps_to_show = ['Original mask', 'Slice mask']
start_gui(data=data, config=config, app_exec=False)
def _start_maps_visualizer(self):
folder = self.selectedOutputFolder.text(
) + '/' + self._get_current_model_name()
maps = mdt.load_volume_maps(folder)
a_map = maps[list(maps.keys())[0]]
config = MapPlotConfig()
config.dimension = 2
if len(a_map.shape) > 2:
config.slice_index = a_map.shape[2] // 2
start_gui(data=SimpleDataInfo.from_paths([folder]),
config=config,
app_exec=False)
def _open_new_directory(self):
initial_dir = self._controller.get_data().directory
new_dir = QFileDialog(self).getExistingDirectory(
caption='Select a folder', directory=initial_dir)
if new_dir:
data = DataInfo.from_dir(new_dir)
config = MapPlotConfig()
if len(data.maps):
config.slice_index = data.get_max_slice_index(
config.dimension) // 2
if self._controller.get_data().maps:
start_gui(data, config, app_exec=False)
else:
self._controller.set_data(data, config)
def view_mask(self):
mask = np.expand_dims(load_brain_mask(self.selectedOutputText.text()),
axis=3)
image_data = load_nifti(self.selectedImageText.text()).get_data()
masked_image = image_data * mask
data = SimpleDataInfo({
'Masked': masked_image,
'DWI': image_data,
'Mask': mask
})
config = MapPlotConfig()
config.dimension = 2
config.slice_index = image_data.shape[2] // 2
config.maps_to_show = ['DWI', 'Masked', 'Mask']
start_gui(data=data, config=config, app_exec=False)
def view_maps(self):
if self._folder is None:
return
maps_to_show = []
for item in [
self.selectMaps.item(index)
for index in range(self.selectMaps.count())
]:
if item.isSelected():
maps_to_show.append(item.text())
data = SimpleDataInfo.from_paths([self._folder])
config = MapPlotConfig()
config.maps_to_show = maps_to_show
config.dimension = self.initialDimensionChooser.value()
config.slice_index = self.initialSliceChooser.value()
start_gui(data=data, config=config, app_exec=False)
