Fixed error in TissueSegmentation when running on Image

Author: smistad

source/FAST/Algorithms/TissueSegmentation/TissueSegmentation.cpp

@@ -200,10 +200,8 @@ void TissueSegmentation::runNeuralNetwork(SpatialDataObject::pointer image) {
         //auto output = finish->runAndGetOutputData<Image>();
         addOutputData(0, output);
     } else if(auto patch = std::dynamic_pointer_cast<Image>(image)) {
-        // Input image is a patch
-        input = patch;
         // If patch is smaller then input size of neural network; run as is
-        if(patch->getWidth() < width || patch->getHeight() << height) {
+        if(patch->getWidth() < width || patch->getHeight() < height) {
             segmentation->connect(patch);
             auto output = segmentation->runAndGetOutputData<Image>();
             addOutputData(0, output);
@@ -213,9 +211,14 @@ void TissueSegmentation::runNeuralNetwork(SpatialDataObject::pointer image) {
             segmentation->connect(generator);
 
             auto stitcher = PatchStitcher::create()->connect(segmentation);
+
+            auto output = stitcher->runAndGetOutputData<SpatialDataObject>();
+            do {
+                output = stitcher->runAndGetOutputData<SpatialDataObject>();
+            } while(!output->isLastFrame());
             // Run until finished
-            auto finish = RunUntilFinished::create()->connect(stitcher);
-            auto output = finish->runAndGetOutputData<Image>();
+            //auto finish = RunUntilFinished::create()->connect(stitcher);
+            //auto output = finish->runAndGetOutputData<Image>();
             addOutputData(0, output);
         }
     } else {