37#include "common/Export.h"
38#include "common/StatusCode.h"
82 const unsigned int moduleIdx = 0);
154 const unsigned int moduleIdx);
175 std::unique_ptr<Impl> _pImpl;
#define MMIND_DL_SDK_EXPORT
Definition Export.h:7
Defines the infer engine.
Definition MMindInferEngine.h:49
StatusCode setInferDeviceType(const InferDeviceType type)
Sets the infer device type.
MMindInferEngine & operator=(const MMindInferEngine &rhs)=delete
StatusCode setBatchSize(const std::vector< unsigned int > &batchSize)
Sets the batch size of the model package.
MMindInferEngine & operator=(MMindInferEngine &&rhs)
StatusCode load()
Loads the model into memory.
StatusCode getResults(std::vector< MMindResult > &results)
Gets the model inference result.
std::vector< DeepLearningAlgoType > getDeepLearningAlgoTypes() const
Gets the model type list.
StatusCode moduleResultVisualization(std::vector< MMindImage > &images, const unsigned int moduleIdx)
Draws the model results of the specified index onto the images.
StatusCode infer(const std::vector< MMindImage > &images)
Makes image inference using the model package inference engine.
MMindInferEngine(const MMindInferEngine &rhs)=delete
void release()
Releases the memory of the model package inference engine.
StatusCode setBatchSize(const unsigned int batchSize, const unsigned int moduleIdx=0)
Sets the batch size of the model package.
StatusCode setDeviceId(const unsigned int deviceId)
Sets the device ID.
MMindInferEngine()
Constructs the infer engine.
StatusCode setFloatPrecision(const std::vector< FloatPrecisionType > &floatPrecisionType)
Sets the float precision of the model package.
StatusCode setFloatPrecision(const FloatPrecisionType floatPrecisionType, const unsigned int moduleIdx=0)
Sets the float precision of the model package.
MMindInferEngine(MMindInferEngine &&rhs)
StatusCode resultVisualization(std::vector< MMindImage > &images)
Draws all the model results onto the images.
StatusCode create(const std::string &modelPath)
Creates an infer engine for model package inference.
FloatPrecisionType
Defines the floating-point format.
Definition common.h:39
InferDeviceType
Defines the infer device type.
Definition common.h:47
StatusCode
Defines the status codes.
Definition StatusCode.h:41