6fe60e43efbac200052a76b0a849c4c7cd4654a5,ilastik/applets/networkClassification/nnClassGui.py,NNClassGui,pred_nn,#NNClassGui#,156

Before Change

        // result = self.classifier_factories[classifier_key].predict_probabilities_pixelwise(self.topLevelOperator.InputImage.value, roi, axistags)
        self.topLevelOperator.Classifier.setValue(self.classifier_factories[classifier_key])

        predict = self.topLevelOperator.CachedPredictionProbabilities[:].wait()
        print ("Done")

        // numpy.save("/Users/jmassa/Desktop/predictions",predict)

After Change

            input_shape = input_shape[1:]
            input_shape = numpy.append(input_shape,None)

            halo_size = self.classifiers[classifier_key].HALO_SIZE
            input_shape[1:3] -= 2*halo_size

            self.topLevelOperator.BlockShape.setValue(input_shape)
            self.topLevelOperator.NumClasses.setValue(3)

Instances