def initializeColorNodes(self):
from mpReview import mpReviewLogic
self.mpReviewColorNode, self.structureNames = mpReviewLogic.loadColorTable(
self.getSetting("Color_File_Name"))
self.segmentedColorName = self.getSetting("Segmentation_Color_Name")
self.segmentedLabelValue = self.mpReviewColorNode.GetColorIndexByName(
self.segmentedColorName)
def runAutomaticSegmentation(self):
logic = AutomaticSegmentationLogic()
logic.addEventObserver(logic.DeepLearningFinishedEvent,
self.onSegmentationFinished)
logic.addEventObserver(
logic.DeepLearningFailedEvent, lambda c, e: self.
onPreopLoadingFailed("Automatic segmentation failed."))
customStatusProgressBar = CustomStatusProgressbar()
customStatusProgressBar.text = "Running DeepInfer for automatic prostate segmentation"
from mpReview import mpReviewLogic
mpReviewColorNode, _ = mpReviewLogic.loadColorTable(
self.getSetting("Color_File_Name", moduleName=self.MODULE_NAME))
domain = 'BWH_WITHOUT_ERC'
prompt = SliceWidgetConfirmYesNoDialog(
self.data.initialVolume,
"Was an endorectal coil used during preop acquisition?").exec_()
self.preopData.usedERC = False
if prompt == qt.QDialogButtonBox.Yes:
self.preopData.usedERC = True
domain = 'BWH_WITH_ERC'
elif prompt == qt.QDialogButtonBox.Cancel:
raise PreProcessedDataError
self.segmentationData.startTime = self.getTime()
logic.run(self.data.initialVolume, domain, mpReviewColorNode)