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 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)