Commit aacfc4cd authored by shizhm's avatar shizhm
Browse files

完善文档

parent 8e6414e6
...@@ -74,40 +74,23 @@ ErrorCode DetectorYOLOV5::Initialize(InitializationParameterOfDetector initializ ...@@ -74,40 +74,23 @@ ErrorCode DetectorYOLOV5::Initialize(InitializationParameterOfDetector initializ
- 归一化[0.0, 1.0] - 归一化[0.0, 1.0]
- 输入数据的尺寸变换:静态推理将输入大小变换为[1,3,608,608],动态推理对输入图像尺寸不做变换。 - 输入数据的尺寸变换:静态推理将输入大小固定为relInputShape=[1,3,608,608],动态推理对输入图像尺寸变换为设定的动态尺寸
``` ```
ErrorCode DetectorYOLOV5::Detect(const cv::Mat &srcImage, std::vector<std::size_t> &relInputShape, std::vector<ResultOfDetection> &resultsOfDetection) ErrorCode DetectorYOLOV5::Detect(const cv::Mat &srcImage, std::vector<std::size_t> &relInputShape, std::vector<ResultOfDetection> &resultsOfDetection, bool dynamic)
{ {
... ...
// 数据预处理 // 数据预处理并转换为NCHW格式
inputSize = cv::Size(relInputShape[3], relInputShape[2]);
cv::Mat inputBlob; cv::Mat inputBlob;
std::vector<std::size_t> relInputShape; cv::dnn::blobFromImage(srcImage,
int height, width;
if(dynamic)
{
width = srcImage.rows;
height = srcImage.cols;
relInputShape = {1,3,height,width};
cv::dnn::blobFromImage(srcImage,
inputBlob,
1 / 255.0,
cv::Size(width, height),
cv::Scalar(0, 0, 0),
true,
false);
}
else
{
cv::dnn::blobFromImage(srcImage,
inputBlob, inputBlob,
1 / 255.0, 1 / 255.0,
inputSize, inputSize,
cv::Scalar(0, 0, 0), cv::Scalar(0, 0, 0),
true, true,
false); false);
}
... ...
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment