Method run

tests/nonconfig_workflow.py:220–226

(self)

        self.postprocessing = Compose([Activationsd(keys="pred", softmax=True), AsDiscreted(keys="pred", argmax=True)])

    def run(self):
        data = self.dataset[0]
        inputs = data[CommonKeys.IMAGE].unsqueeze(0).to(self.device)
        self.net.eval()
        with torch.no_grad():
            data[CommonKeys.PRED] = self.inferer(inputs, self.net)
        self.dataflow.update({CommonKeys.PRED: self.postprocessing(data)[CommonKeys.PRED]})

    def finalize(self):
        pass