"src/vscode:/vscode.git/clone" did not exist on "4c4b323c1ff5f4cece9b115e60b21655ed551127"
Commit aacfc4cd authored by shizhm's avatar shizhm
Browse files

完善文档

parent 8e6414e6
......@@ -74,40 +74,23 @@ ErrorCode DetectorYOLOV5::Initialize(InitializationParameterOfDetector initializ
- 归一化[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;
std::vector<std::size_t> relInputShape;
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,
cv::dnn::blobFromImage(srcImage,
inputBlob,
1 / 255.0,
inputSize,
cv::Scalar(0, 0, 0),
true,
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