fastdeploy::vision::matting::RobustVideoMatting Class Reference

RobustVideoMatting model object used when to load a RobustVideoMatting model exported by RobustVideoMatting. More...

#include <rvm.h>

Inheritance diagram for fastdeploy::vision::matting::RobustVideoMatting:

Collaboration diagram for fastdeploy::vision::matting::RobustVideoMatting:

Public Member Functions
	RobustVideoMatting (const std::string &model_file, const std::string &params_file="", const RuntimeOption &custom_option=RuntimeOption(), const ModelFormat &model_format=ModelFormat::ONNX)
	Set path of model file and configuration file, and the configuration of runtime. More...
std::string	ModelName () const
	Get model's name.
bool	Predict (cv::Mat *im, MattingResult *result)
	Predict the matting result for an input image. More...
Public Member Functions inherited from fastdeploy::FastDeployModel
virtual bool	Infer (std::vector< FDTensor > &input_tensors, std::vector< FDTensor > *output_tensors)
	Inference the model by the runtime. This interface is included in the Predict() function, so we don't call Infer() directly in most common situation.
virtual bool	Infer ()
	Inference the model by the runtime. This interface is using class member reused_input_tensors_ to do inference and writing results to reused_output_tensors_.
virtual int	NumInputsOfRuntime ()
	Get number of inputs for this model.
virtual int	NumOutputsOfRuntime ()
	Get number of outputs for this model.
virtual TensorInfo	InputInfoOfRuntime (int index)
	Get input information for this model.
virtual TensorInfo	OutputInfoOfRuntime (int index)
	Get output information for this model.
virtual bool	Initialized () const
	Check if the model is initialized successfully.
virtual void	EnableRecordTimeOfRuntime ()
	This is a debug interface, used to record the time of runtime (backend + h2d + d2h) More...
virtual void	DisableRecordTimeOfRuntime ()
	Disable to record the time of runtime, see EnableRecordTimeOfRuntime() for more detail.
virtual std::map< std::string, float >	PrintStatisInfoOfRuntime ()
	Print the statistic information of runtime in the console, see function EnableRecordTimeOfRuntime() for more detail.
virtual bool	EnabledRecordTimeOfRuntime ()
	Check if the EnableRecordTimeOfRuntime() method is enabled.
virtual double	GetProfileTime ()
	Get profile time of Runtime after the profile process is done.
virtual void	ReleaseReusedBuffer ()
	Release reused input/output buffers.

Public Attributes
std::vector< int >	size
	Preprocess image size, the default is (1080, 1920)
bool	video_mode
	Whether to open the video mode, if there are some irrelevant pictures, set it to fasle, the default is true // NOLINT.
bool	swap_rb
	Whether convert to RGB, Set to false if you have converted YUV format images to RGB outside the model, dafault true // NOLINT.
Public Attributes inherited from fastdeploy::FastDeployModel
std::vector< Backend >	valid_cpu_backends = {Backend::ORT}
	Model's valid cpu backends. This member defined all the cpu backends have successfully tested for the model.
std::vector< Backend >	valid_gpu_backends = {Backend::ORT}
std::vector< Backend >	valid_ipu_backends = {}
std::vector< Backend >	valid_timvx_backends = {}
std::vector< Backend >	valid_directml_backends = {}
std::vector< Backend >	valid_ascend_backends = {}
std::vector< Backend >	valid_kunlunxin_backends = {}
std::vector< Backend >	valid_rknpu_backends = {}
std::vector< Backend >	valid_sophgonpu_backends = {}

Detailed Description

RobustVideoMatting model object used when to load a RobustVideoMatting model exported by RobustVideoMatting.

Constructor & Destructor Documentation

RobustVideoMatting()

fastdeploy::vision::matting::RobustVideoMatting::RobustVideoMatting	(	const std::string &	model_file,
		const std::string &	params_file = "",
		const RuntimeOption &	custom_option = RuntimeOption(),
		const ModelFormat &	model_format = ModelFormat::ONNX
	)

Set path of model file and configuration file, and the configuration of runtime.

Parameters

[in]	model_file	Path of model file, e.g rvm/rvm_mobilenetv3_fp32.onnx
[in]	params_file	Path of parameter file, if the model format is ONNX, this parameter will be ignored
[in]	custom_option	RuntimeOption for inference, the default will use cpu, and choose the backend defined in valid_cpu_backends
[in]	model_format	Model format of the loaded model, default is ONNX format

Member Function Documentation

Predict()

bool fastdeploy::vision::matting::RobustVideoMatting::Predict	(	cv::Mat *	im,
		MattingResult *	result
	)

Predict the matting result for an input image.

Parameters

[in]	im	The input image data, comes from cv::imread()
[in]	result	The output matting result will be writen to this structure

Returns

true if the prediction successed, otherwise false

---

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

• /fastdeploy/my_work/FastDeploy/fastdeploy/vision/matting/contrib/rvm.h
• /fastdeploy/my_work/FastDeploy/fastdeploy/vision/matting/contrib/rvm.cc