Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
YOLOv8_migraphx
Commits
e155e36a
"src/vscode:/vscode.git/clone" did not exist on "0b7b979e71cdd2b02a72c3112217f26b385c26b3"
Commit
e155e36a
authored
Dec 14, 2023
by
yaoht
Browse files
fix README.md
parent
e7dd6859
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
Doc/Tutorial_Cpp.md
Doc/Tutorial_Cpp.md
+2
-2
No files found.
Doc/Tutorial_Cpp.md
View file @
e155e36a
...
...
@@ -96,7 +96,7 @@ ErrorCode DetectorYOLOV8::Detect(const cv::Mat &srcImage, std::vector<std::size_
## 推理
完成图像预处理以及YOLOV8目标检测相关参数设置之后开始执行推理,
利用migraphx推理
计算
得到YOLOV8模型的输出。其中静态推理输入数据inputData的shape大小为模型的固定输入尺寸,动态推理则为实际输入的尺寸。
利用migraphx推理得到YOLOV8模型的输出。其中静态推理输入数据inputData的shape大小为模型的固定输入尺寸,动态推理则为实际输入的尺寸。
```
cpp
ErrorCode
DetectorYOLOV8
::
Detect
(
const
cv
::
Mat
&
srcImage
,
std
::
vector
<
std
::
size_t
>
&
relInputShape
,
std
::
vector
<
ResultOfDetection
>
&
resultsOfDetection
,
bool
dynamic
)
...
...
@@ -137,7 +137,7 @@ std::vector<migraphx::argument> inferenceResults = net.eval(inputData, outputNam
如果没有指定outputName参数,则默认输出所有输出节点,此时输出节点的顺序与ONNX中输出节点顺序保持一致,可以通过netron查看ONNX文件的输出节点的顺序。
获取上述信息之后进行proposal筛选,筛选过程根据confidenceThreshold阈值进行筛选,
当
proposal的最大置信度得分maxClassScore
大于该
阈值,则进一步获取当前proposal的坐标信息和预测物体类别信息,小于该阈值则不做处理。
获取上述信息之后进行proposal筛选,筛选过程根据confidenceThreshold阈值进行筛选,proposal的最大置信度得分maxClassScore
由80个类别的分数最高值确定,当maxClassScore大于confidenceThreshold
阈值,则进一步获取当前proposal的坐标信息和预测物体类别信息,小于该阈值则不做处理。
```
cpp
ErrorCode
DetectorYOLOV5
::
Detect
(
const
cv
::
Mat
&
srcImage
,
std
::
vector
<
std
::
size_t
>
&
relInputShape
,
std
::
vector
<
ResultOfDetection
>
&
resultsOfDetection
,
bool
dynamic
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment