opencv_convert.h

Go to the documentation of this file.

#ifndef _NNDEPLOY_PREPROCESS_OPENCV_CONVERT_H_
#define _NNDEPLOY_PREPROCESS_OPENCV_CONVERT_H_
 
#include "nndeploy/base/any.h"
#include "nndeploy/base/common.h"
#include "nndeploy/base/glic_stl_include.h"
#include "nndeploy/base/log.h"
#include "nndeploy/base/macro.h"
#include "nndeploy/base/object.h"
#include "nndeploy/base/opencv_include.h"
#include "nndeploy/base/status.h"
#include "nndeploy/base/string.h"
#include "nndeploy/dag/edge.h"
#include "nndeploy/dag/node.h"
#include "nndeploy/device/buffer.h"
#include "nndeploy/device/device.h"
#include "nndeploy/device/memory_pool.h"
#include "nndeploy/device/tensor.h"
#include "nndeploy/preprocess/params.h"
 
namespace nndeploy {
namespace preprocess {
 
class NNDEPLOY_CC_API OpenCvConvert {
 public:
  static int convertFromCvtColorType(base::CvtColorType src);
 
  static int convertFromInterpType(base::InterpType src);
 
  static int convertFromBorderType(base::BorderType src);
 
  static cv::Scalar convertFromScalar(const base::Scalar2d &src);
 
  static int convertFromDataType(base::DataType src);
 
  static bool normalize(const cv::Mat &src, cv::Mat &dst,
                        base::DataType data_type, float *scale, float *mean,
                        float *std);
 
  static bool convertToTensor(const cv::Mat &src, device::Tensor *dst,
                              bool normalize, float *scale, float *mean,
                              float *std);
  static bool convertToBatchTensor(const cv::Mat &src, device::Tensor *dst,
                                bool normalize, float *scale, float *mean,
                                float *std, int batch);
};
 
}  // namespace preprocess
}  // namespace nndeploy
 
#endif