FAST  3.2.0
Framework for Heterogeneous Medical Image Computing and Visualization
Public Member Functions | List of all members
fast::BoundingBoxNetwork Class Reference

#include <BoundingBoxNetwork.hpp>

+ Inheritance diagram for fast::BoundingBoxNetwork:
+ Collaboration diagram for fast::BoundingBoxNetwork:

Public Member Functions

void setThreshold (float threshold)
 
void loadAttributes () override
 
void setAnchors (std::vector< std::vector< Vector2f >> anchors)
 
- Public Member Functions inherited from fast::NeuralNetwork
void load (std::string filename)
 
void load (std::vector< uint8_t > model, std::vector< uint8_t > weights)
 
void setInferenceEngine (InferenceEngine::pointer engine)
 
void setInferenceEngine (std::string engine)
 
InferenceEngine::pointer getInferenceEngine () const
 
void setInputNode (uint portID, std::string name, NodeType type=NodeType::IMAGE, TensorShape shape={})
 
void setOutputNode (uint portID, std::string name, NodeType type=NodeType::IMAGE, TensorShape shape={})
 
void setScaleFactor (float scale)
 
void setMeanAndStandardDeviation (float mean, float std)
 
void setMinAndMaxIntensity (float min, float max)
 
void setSignedInputNormalization (bool signedInputNormalization)
 
void setPreserveAspectRatio (bool preserve)
 
void setHorizontalFlipping (bool flip)
 
void setTemporalWindow (uint window)
 
virtual void setInputSize (std::string name, std::vector< int > size)
 
void loadAttributes ()
 
virtual ~NeuralNetwork ()
 
- Public Member Functions inherited from fast::ProcessObject
virtual ~ProcessObject ()
 
void update (int executeToken=-1)
 
RuntimeMeasurement::pointer getRuntime ()
 
RuntimeMeasurement::pointer getRuntime (std::string name)
 
RuntimeMeasurementsManager::pointer getAllRuntimes ()
 
void enableRuntimeMeasurements ()
 
void disableRuntimeMeasurements ()
 
void setMainDevice (ExecutionDevice::pointer device)
 
void setMainDeviceCriteria (const DeviceCriteria &citeria)
 
ExecutionDevice::pointer getMainDevice () const
 
void setDevice (uint deviceNumber, ExecutionDevice::pointer device)
 
void setDeviceCriteria (uint deviceNumber, const DeviceCriteria &criteria)
 
ExecutionDevice::pointer getDevice (uint deviceNumber) const
 
virtual DataChannel::pointer getOutputPort (uint portID=0)
 
virtual DataChannel::pointer getInputPort (uint portID=0)
 
virtual void setInputConnection (DataChannel::pointer port)
 
virtual void setInputConnection (uint portID, DataChannel::pointer port)
 
virtual void setInputData (DataObject::pointer data)
 
virtual void setInputData (uint portID, DataObject::pointer data)
 
int getNrOfInputConnections () const
 
int getNrOfOutputPorts () const
 
virtual std::string getNameOfClass () const =0
 
std::shared_ptr< AttributegetAttribute (std::string id)
 
std::unordered_map< std::string, std::shared_ptr< Attribute > > getAttributes ()
 
void setAttributes (std::vector< std::shared_ptr< Attribute >> attributes)
 
void stopPipeline ()
 
void setModified (bool modified)
 
template<class DataType >
std::shared_ptr< DataTypeupdateAndGetOutputData (uint portID=0)
 
- Public Member Functions inherited from fast::Object
 Object ()
 
virtual ~Object ()
 
ReportergetReporter ()
 

Additional Inherited Members

- Public Types inherited from fast::ProcessObject
typedef std::shared_ptr< ProcessObjectpointer
 
- Public Types inherited from fast::Object
typedef std::shared_ptr< Objectpointer
 
- Static Public Member Functions inherited from fast::ProcessObject
static std::string getStaticNameOfClass ()
 
- Static Public Member Functions inherited from fast::Object
static std::string getStaticNameOfClass ()
 
- Protected Member Functions inherited from fast::NeuralNetwork
 NeuralNetwork ()
 
virtual void run ()
 
std::unordered_map< std::string, Tensor::pointerprocessInputData ()
 
std::vector< std::shared_ptr< Image > > resizeImages (const std::vector< std::shared_ptr< Image >> &images, int width, int height, int depth)
 
Tensor::pointer convertImagesToTensor (std::vector< std::shared_ptr< Image >> image, const TensorShape &shape, bool temporal)
 
Tensor::pointer standardizeOutputTensorData (Tensor::pointer tensor, int sample=0)
 
- Protected Member Functions inherited from fast::ProcessObject
 ProcessObject ()
 
virtual void preExecute ()
 
virtual void postExecute ()
 
template<class DataType >
void createInputPort (uint portID, bool required=true)
 
template<class DataType >
void createOutputPort (uint portID)
 
template<class DataType >
std::shared_ptr< DataTypegetInputData (uint portID=0)
 
template<class DataType >
std::shared_ptr< DataTypegetOutputData (uint portID=0)
 
void addOutputData (uint portID, DataObject::pointer data)
 
bool hasNewInputData (uint portID)
 
virtual void waitToFinish ()
 
void createOpenCLProgram (std::string sourceFilename, std::string name="")
 
cl::Program getOpenCLProgram (std::shared_ptr< OpenCLDevice > device, std::string name="", std::string buildOptions="")
 
void createFloatAttribute (std::string id, std::string name, std::string description, float initialValue)
 
void createIntegerAttribute (std::string id, std::string name, std::string description, int initialValue)
 
void createBooleanAttribute (std::string id, std::string name, std::string description, bool initialValue)
 
void createStringAttribute (std::string id, std::string name, std::string description, std::string initialValue)
 
float getFloatAttribute (std::string id)
 
int getIntegerAttribute (std::string id)
 
bool getBooleanAttribute (std::string id)
 
std::string getStringAttribute (std::string id)
 
std::vector< float > getFloatListAttribute (std::string id)
 
std::vector< int > getIntegerListAttribute (std::string id)
 
std::vector< bool > getBooleanListAttribute (std::string id)
 
std::vector< std::string > getStringListAttribute (std::string id)
 
void changeDeviceOnInputs (uint deviceNumber, ExecutionDevice::pointer device)
 
void validateInputPortExists (uint portID)
 
void validateOutputPortExists (uint portID)
 
- Protected Member Functions inherited from fast::Object
ReporterreportError ()
 
ReporterreportWarning ()
 
ReporterreportInfo ()
 
ReporterEnd reportEnd () const
 
- Protected Attributes inherited from fast::NeuralNetwork
bool mPreserveAspectRatio
 
bool mHorizontalImageFlipping = false
 
bool mSignedInputNormalization = false
 
int mTemporalWindow = 0
 
int m_batchSize
 
float mScaleFactor
 
float mMean
 
float mStd
 
float mMinIntensity
 
float mMaxIntensity
 
bool mMinAndMaxIntensitySet = false
 
Vector3f mNewInputSpacing
 
std::unordered_map< std::string, std::vector< int > > mInputSizes
 
std::unordered_map< int, DataObject::pointerm_processedOutputData
 
std::shared_ptr< InferenceEnginem_engine
 
std::unordered_map< std::string, std::vector< std::shared_ptr< Image > > > mInputImages
 
std::unordered_map< std::string, std::vector< std::shared_ptr< Tensor > > > mInputTensors
 
- Protected Attributes inherited from fast::ProcessObject
bool mIsModified
 
int m_lastExecuteToken = -1
 
RuntimeMeasurementsManager::pointer mRuntimeManager
 
std::unordered_map< uint, bool > mRequiredInputs
 
std::unordered_map< uint, std::vector< uint > > mInputDevices
 
std::unordered_map< uint, ExecutionDevice::pointermDevices
 
std::unordered_map< uint, DeviceCriteriamDeviceCriteria
 
std::unordered_map< uint, DataChannel::pointermInputConnections
 
std::unordered_map< uint, std::vector< std::weak_ptr< DataChannel > > > mOutputConnections
 
std::unordered_map< uint, bool > mInputPorts
 
std::unordered_set< uintmOutputPorts
 
std::unordered_map< uint, std::pair< DataObject::pointer, uint64_t > > mLastProcessed
 
std::unordered_map< std::string, std::shared_ptr< OpenCLProgram > > mOpenCLPrograms
 
std::unordered_map< std::string, std::shared_ptr< Attribute > > mAttributes
 
std::unordered_map< std::string, std::string > m_frameData
 
std::unordered_set< std::string > m_lastFrame
 
int m_maximumNrOfFrames = -1
 
- Protected Attributes inherited from fast::Object
std::weak_ptr< ObjectmPtr
 

Member Function Documentation

◆ loadAttributes()

void fast::BoundingBoxNetwork::loadAttributes ( )
overridevirtual

Reimplemented from fast::ProcessObject.

◆ setAnchors()

void fast::BoundingBoxNetwork::setAnchors ( std::vector< std::vector< Vector2f >>  anchors)

◆ setThreshold()

void fast::BoundingBoxNetwork::setThreshold ( float  threshold)

The documentation for this class was generated from the following file: