SegmentationNetwork.hpp

Go to the documentation of this file.

#pragma once
 
#include "NeuralNetwork.hpp"
 
namespace fast {
 
class TensorToSegmentation;
 
class FAST_EXPORT SegmentationNetwork : public NeuralNetwork {
    FAST_OBJECT(SegmentationNetwork)
    public:
        void setHeatmapOutput();
        void setSegmentationOutput();
        void setThreshold(float threshold);
        void loadAttributes();
        void setResizeBackToOriginalSize(bool resize);
    private:
        SegmentationNetwork();
        void execute();
        bool mHeatmapOutput = false;
        bool m_resizeBackToOriginalSize = false;
        std::shared_ptr<TensorToSegmentation> m_tensorToSegmentation;
 
};
 
}