-
Notifications
You must be signed in to change notification settings - Fork 101
Example: Neural network segmentation
Erik Smistad edited this page Jun 11, 2018
·
1 revision
/**
* Examples/Segmentation/neuralNetworkSegmentation.cpp
*
* If you edit this example, please also update the wiki and source code file in the repository.
*/
#include <FAST/Visualization/DualViewWindow.hpp>
#include <FAST/Visualization/SegmentationRenderer/SegmentationRenderer.hpp>
#include <FAST/Algorithms/ImageResampler/ImageResampler.hpp>
#include "FAST/Testing.hpp"
#include "FAST/Streamers/ImageFileStreamer.hpp"
#include "FAST/Visualization/SimpleWindow.hpp"
#include "FAST/Visualization/ImageRenderer/ImageRenderer.hpp"
#include "FAST/Algorithms/NeuralNetwork/PixelClassifier.hpp"
using namespace fast;
int main() {
Reporter::setGlobalReportMethod(Reporter::COUT);
ImageFileStreamer::pointer streamer = ImageFileStreamer::New();
streamer->setFilenameFormat(Config::getTestDataPath() + "US/JugularVein/US-2D_#.mhd");
streamer->setTimestampFilename(Config::getTestDataPath() + "US/JugularVein/timestamps.fts");
PixelClassifier::pointer segmentation = PixelClassifier::New();
segmentation->setNrOfClasses(3);
segmentation->load(Config::getTestDataPath() + "NeuralNetworkModels/jugular_vein_segmentation.pb");
segmentation->setInputSize(256, 256);
segmentation->setScaleFactor(1.0f / 255.0f);
segmentation->setOutputParameters({"conv2d_23/truediv"});
segmentation->setInputConnection(streamer->getOutputPort());
SegmentationRenderer::pointer segmentationRenderer = SegmentationRenderer::New();
segmentationRenderer->addInputConnection(segmentation->getOutputPort());
segmentationRenderer->setOpacity(0.25);
segmentationRenderer->setColor(Segmentation::LABEL_FOREGROUND, Color::Red());
segmentationRenderer->setColor(Segmentation::LABEL_BLOOD, Color::Blue());
ImageRenderer::pointer imageRenderer = ImageRenderer::New();
imageRenderer->setInputConnection(streamer->getOutputPort());
SimpleWindow::pointer window = SimpleWindow::New();
window->addRenderer(imageRenderer);
window->addRenderer(segmentationRenderer);
window->set2DMode();
window->getView()->setBackgroundColor(Color::Black());
window->start();
}
If this wiki page lacks some information or is incorrect please let us know! You can edit this wiki page yourself, send an email to [email protected] or