cv::dnn::SegmentationModel Class Reference

This class represents high-level API for segmentation models. More...

#include <opencv2/dnn/dnn.hpp>

Inheritance diagram for cv::dnn::SegmentationModel:

Public Member Functions
	SegmentationModel (const String &model, const String &config="")
	Create segmentation model from network represented in one of the supported formats. An order of model and config arguments does not matter. More...
	SegmentationModel (const Net &network)
	Create model from deep learning network. More...
void	segment (InputArray frame, OutputArray mask)
	Given the input frame, create input blob, run net. More...
Public Member Functions inherited from cv::dnn::Model
	Model ()
	Model (const Model &)=default
	Model (Model &&)=default
	Model (const String &model, const String &config="")
	Create model from deep learning network represented in one of the supported formats. An order of model and config arguments does not matter. More...
	Model (const Net &network)
	Create model from deep learning network. More...
Impl *	getImpl () const
Impl &	getImplRef () const
Net &	getNetwork_ () const
Net &	getNetwork_ ()
	operator Net & () const
Model &	operator= (const Model &)=default
Model &	operator= (Model &&)=default
void	predict (InputArray frame, OutputArrayOfArrays outs) const
	Given the input frame, create input blob, run net and return the output blobs. More...
Model &	setInputCrop (bool crop)
	Set flag crop for frame. More...
Model &	setInputMean (const Scalar &mean)
	Set mean value for frame. More...
void	setInputParams (double scale=1.0, const Size &size=Size(), const Scalar &mean=Scalar(), bool swapRB=false, bool crop=false)
	Set preprocessing parameters for frame. More...
Model &	setInputScale (const Scalar &scale)
	Set scalefactor value for frame. More...
Model &	setInputSize (const Size &size)
	Set input size for frame. More...
Model &	setInputSize (int width, int height)
Model &	setInputSwapRB (bool swapRB)
	Set flag swapRB for frame. More...
Model &	setPreferableBackend (dnn::Backend backendId)
Model &	setPreferableTarget (dnn::Target targetId)

Additional Inherited Members
Protected Attributes inherited from cv::dnn::Model
Ptr< Impl >	impl

Detailed Description

This class represents high-level API for segmentation models.

SegmentationModel allows to set params for preprocessing input image. SegmentationModel creates net from file with trained weights and config, sets preprocessing input, runs forward pass and returns the class prediction for each pixel.

Constructor & Destructor Documentation

SegmentationModel() [1/2]

cv::dnn::SegmentationModel::SegmentationModel	(	const String &	model,
		const String &	config = ""
	)

Create segmentation model from network represented in one of the supported formats. An order of model and config arguments does not matter.

Parameters

[in]	model	Binary file contains trained weights.
[in]	config	Text file contains network configuration.

SegmentationModel() [2/2]

cv::dnn::SegmentationModel::SegmentationModel

(

const Net &

network

)

Create model from deep learning network.

Parameters

[in]

network

Net object.

Member Function Documentation

segment()

void cv::dnn::SegmentationModel::segment	(	InputArray	frame,
		OutputArray	mask
	)

Python:
	cv.dnn.SegmentationModel.segment(	frame[, mask]	) ->	mask

Given the input frame, create input blob, run net.

Parameters

[in]	frame	The input image.
[out]	mask	Allocated class prediction for each pixel

---

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

• opencv2/dnn/dnn.hpp