FAST  3.2.0
Framework for Heterogeneous Medical Image Computing and Visualization
Classes | Namespaces | Macros | Enumerations | Functions
InferenceEngine.hpp File Reference
#include "FAST/Data/DataTypes.hpp"
#include <FAST/Data/Tensor.hpp>
#include <FAST/Data/TensorShape.hpp>
+ Include dependency graph for InferenceEngine.hpp:
+ This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  fast::InferenceDeviceInfo
 
class  fast::InferenceEngine
 
struct  fast::InferenceEngine::NetworkNode
 

Namespaces

 fast
 

Macros

#define DEFINE_INFERENCE_ENGINE(classType, exportStatement)
 

Enumerations

enum  fast::ImageOrdering { fast::ImageOrdering::ChannelFirst, fast::ImageOrdering::ChannelLast }
 
enum  fast::NodeType { fast::NodeType::IMAGE, fast::NodeType::TENSOR }
 
enum  fast::InferenceDeviceType {
  fast::InferenceDeviceType::ANY, fast::InferenceDeviceType::CPU, fast::InferenceDeviceType::GPU, fast::InferenceDeviceType::VPU,
  fast::InferenceDeviceType::OTHER
}
 
enum  fast::ModelFormat {
  fast::ModelFormat::PROTOBUF, fast::ModelFormat::SAVEDMODEL, fast::ModelFormat::ONNX, fast::ModelFormat::OPENVINO,
  fast::ModelFormat::UFF
}
 

Functions

FAST_EXPORT std::string fast::getModelFileExtension (ModelFormat format)
 
FAST_EXPORT ModelFormat fast::getModelFormat (std::string filename)
 
FAST_EXPORT std::string fast::getModelFormatName (ModelFormat format)
 

Macro Definition Documentation

◆ DEFINE_INFERENCE_ENGINE

#define DEFINE_INFERENCE_ENGINE (   classType,
  exportStatement 
)
Value:
extern "C" exportStatement \
InferenceEngine* load() { \
return new classType(); \
} \