Utils.H File Reference

#include <map>
#include <string>
#include <opencv2/core/core.hpp>
#include <tensorflow/lite/c/common.h>
#include <ovxlib/vsi_nn_pub.h>
#include <hailo/hailort.h>
#include <onnxruntime_cxx_api.h>

Include dependency graph for Utils.H:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	jevois
	jevois::dnn

Functions
std::map< int, std::string >	jevois::dnn::readLabelsFile (std::string const &fname)
	Read a label file. More...
std::string	jevois::dnn::getLabel (std::map< int, std::string > const &labels, int id)
	Get a label from an id. More...
int	jevois::dnn::stringToRGBA (std::string const &label, unsigned char alpha=128)
	Compute a color from a label name. More...
void	jevois::dnn::topK (float const *pfProb, float *pfMaxProb, uint32_t *pMaxClass, uint32_t outputCount, uint32_t topNum)
	Get top-k entries and their indices. More...
std::string	jevois::dnn::shapestr (cv::Mat const &m)
	Get a string of the form: "nD AxBxC... TYPE" from an n-dimensional cv::Mat with data type TYPE. More...
std::string	jevois::dnn::shapestr (TfLiteTensor const *t)
	Get a string of the form: "nD AxBxC... TYPE" from an n-dimensional TfLiteTensor with data type TYPE. More...
std::string	jevois::dnn::shapestr (vsi_nn_tensor_attr_t const &attr)
	Get a string of the form: "nD AxBxC... TYPE" from an n-dimensional NPU tensor with data type TYPE. More...
std::vector< size_t >	jevois::dnn::strshape (std::string const &str)
	Get a vector of size_t from a string containing AxBxC... More...
int	jevois::dnn::tf2cv (TfLiteType t)
	Convert from TensorFlow data type to OpenCV. More...
vsi_nn_type_e	jevois::dnn::tf2vsi (TfLiteType t)
	Convert from TensorFlow data type to vsi_nn. More...
int	jevois::dnn::vsi2cv (vsi_nn_type_e t)
	Convert from NPU data type to OpenCV. More...
void	jevois::dnn::clamp (cv::Rect &r, int width, int height)
	Clamp a rectangle to within given image width and height. More...
void	jevois::dnn::clamp (cv::Rect2f &r, float width, float height)
	Clamp a rectangle to within given image width and height. More...
std::vector< vsi_nn_tensor_attr_t >	jevois::dnn::parseTensorSpecs (std::string const &specs)
	Parse tensor specification. More...
cv::Mat	jevois::dnn::attrmat (vsi_nn_tensor_attr_t const &attr, void *dataptr=nullptr)
	Construct a cv::Mat from attr and possibly data pointer. More...
std::vector< int >	jevois::dnn::attrdims (vsi_nn_tensor_attr_t const &attr)
	Get a tensor dims as a vector of int, useful to construct a matching cv::Mat. More...
cv::Size	jevois::dnn::attrsize (vsi_nn_tensor_attr_t const &attr)
	Get a tensor's (width, height) size in cv::Size format, skipping over other dimensions. More...
std::string	jevois::dnn::attrstr (vsi_nn_tensor_attr_t const &attr)
	Get a string describing the specs of a tensor, including quantification specs (not provided by shapestr()) More...
bool	jevois::dnn::attrmatch (vsi_nn_tensor_attr_t const &attr, cv::Mat const &blob)
	Check that a cv::Mat blob matches exactly the spec of an attr. More...
vsi_nn_tensor_attr_t	jevois::dnn::tensorattr (TfLiteTensor const *t)
	Get tensor shape and type attributes for a TensorFlow Lite tensor. More...
size_t	jevois::dnn::softmax (float const *input, size_t const n, size_t const stride, float const fac, float *output, bool maxonly)
	Apply softmax to a float vector. More...
cv::Mat	jevois::dnn::quantize (cv::Mat const &m, vsi_nn_tensor_attr_t const &attr)
	Quantize from float32 to fixed-point according to the quantization spec in attr. More...
cv::Mat	jevois::dnn::dequantize (cv::Mat const &m, vsi_nn_tensor_attr_t const &attr)
	Dequantize an output to float32 according to the quantization spec in attr. More...
size_t	jevois::dnn::effectiveDims (cv::Mat const &m)
	Returns the number of non-unit dims in a cv::Mat. More...
std::string	jevois::dnn::shapestr (hailo_vstream_info_t const &vi)
	Get a string of the form: "nD AxBxC... TYPE" from an n-dimensional Hailo tensor with data type TYPE. More...
vsi_nn_tensor_attr_t	jevois::dnn::tensorattr (hailo_vstream_info_t const &vi)
	Get tensor shape and type attributes for a Hailo tensor. More...
vsi_nn_type_e	jevois::dnn::hailo2vsi (hailo_format_type_t t)
	Convert from Hailo data type to vsi_nn. More...
vsi_nn_type_e	jevois::dnn::onnx2vsi (ONNXTensorElementDataType t)
	Convert from ONNX-Runtime data type to vsi_nn. More...
std::string	jevois::dnn::shapestr (Ort::ConstTensorTypeAndShapeInfo const &ti)
	Get a string of the form: "nD AxBxC... TYPE" from an n-dimensional ONNX tensor with data type TYPE. More...
vsi_nn_tensor_attr_t	jevois::dnn::tensorattr (Ort::ConstTensorTypeAndShapeInfo const &ti)
	Get tensor shape and type attributes for an ONNX-runtime tensor. More...