Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleX
提交
4824363c
P
PaddleX
项目概览
PaddlePaddle
/
PaddleX
通知
138
Star
4
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
43
列表
看板
标记
里程碑
合并请求
5
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleX
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
43
Issue
43
列表
看板
标记
里程碑
合并请求
5
合并请求
5
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
4824363c
编写于
8月 27, 2020
作者:
S
syyxsxx
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add namespace InferenceEngine::
上级
3365478f
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
8 addition
and
11 deletion
+8
-11
deploy/openvino/src/paddlex.cpp
deploy/openvino/src/paddlex.cpp
+8
-11
未找到文件。
deploy/openvino/src/paddlex.cpp
浏览文件 @
4824363c
...
...
@@ -32,14 +32,14 @@ void Model::create_predictor(const std::string& model_dir,
for
(
const
auto
&
inputInfoItem
:
inputInfo
)
{
if
(
inputInfoItem
.
second
->
getTensorDesc
().
getDims
().
size
()
==
4
)
{
imageInputName
=
inputInfoItem
.
first
;
inputInfoItem
.
second
->
setPrecision
(
Precision
::
FP32
);
inputInfoItem
.
second
->
setPrecision
(
InferenceEngine
::
Precision
::
FP32
);
inputInfoItem
.
second
->
getPreProcess
().
setResizeAlgorithm
(
RESIZE_BILINEAR
);
inputInfoItem
.
second
->
setLayout
(
Layout
::
NCHW
);
InferenceEngine
::
RESIZE_BILINEAR
);
inputInfoItem
.
second
->
setLayout
(
InferenceEngine
::
Layout
::
NCHW
);
}
if
(
inputInfoItem
.
second
->
getTensorDesc
().
getDims
().
size
()
==
2
)
{
imageInputName
=
inputInfoItem
.
first
;
inputInfoItem
.
second
->
setPrecision
(
Precision
::
FP32
);
inputInfoItem
.
second
->
setPrecision
(
InferenceEngine
::
Precision
::
FP32
);
}
}
if
(
device
==
"MYRIAD"
)
{
...
...
@@ -114,7 +114,7 @@ bool Model::predict(const cv::Mat& im, ClsResult* result) {
std
::
string
output_name
=
network_
.
getOutputsInfo
().
begin
()
->
first
;
output_
=
infer_request
.
GetBlob
(
output_name
);
InferenceEngine
::
MemoryBlob
::
CPtr
moutput
=
as
<
InferenceEngine
::
MemoryBlob
>
(
output_
);
InferenceEngine
::
as
<
InferenceEngine
::
MemoryBlob
>
(
output_
);
auto
moutputHolder
=
moutput
->
rmap
();
float
*
outputs_data
=
moutputHolder
.
as
<
float
*>
();
...
...
@@ -165,7 +165,7 @@ bool Model::predict(const cv::Mat& im, DetResult* result) {
std
::
string
outputName
=
iter
->
first
;
InferenceEngine
::
Blob
::
Ptr
output
=
infer_request
.
GetBlob
(
outputName
);
InferenceEngine
::
MemoryBlob
::
CPtr
moutput
=
as
<
InferenceEngine
::
MemoryBlob
>
(
output
);
InferenceEngine
::
as
<
InferenceEngine
::
MemoryBlob
>
(
output
);
InferenceEngine
::
TensorDesc
blob_output
=
moutput
->
getTensorDesc
();
std
::
vector
<
size_t
>
output_shape
=
blob_output
.
getDims
();
auto
moutputHolder
=
moutput
->
rmap
();
...
...
@@ -221,9 +221,6 @@ bool Model::predict(const cv::Mat& im, SegResult* result) {
//
infer_request
.
Infer
();
if
(
count_num_
>=
20
)
{
total_time_
=
total_time_
+
time_used
.
count
();
}
OInferenceEngine
::
utputsDataMap
out_map
=
network_
.
getOutputsInfo
();
auto
iter
=
out_map
.
begin
();
...
...
@@ -232,7 +229,7 @@ bool Model::predict(const cv::Mat& im, SegResult* result) {
InferenceEngine
::
Blob
::
Ptr
output_score
=
infer_request
.
GetBlob
(
output_name_score
);
InferenceEngine
::
MemoryBlob
::
CPtr
moutput_score
=
as
<
InferenceEngine
::
MemoryBlob
>
(
output_score
);
InferenceEngine
::
as
<
InferenceEngine
::
MemoryBlob
>
(
output_score
);
InferenceEngine
::
TensorDesc
blob_score
=
moutput_score
->
getTensorDesc
();
std
::
vector
<
size_t
>
output_score_shape
=
blob_score
.
getDims
();
int
size
=
1
;
...
...
@@ -250,7 +247,7 @@ bool Model::predict(const cv::Mat& im, SegResult* result) {
InferenceEngine
::
Blob
::
Ptr
output_label
=
infer_request
.
GetBlob
(
output_name_label
);
InferenceEngine
::
MemoryBlob
::
CPtr
moutput_label
=
as
<
InferenceEngine
::
MemoryBlob
>
(
output_label
);
InferenceEngine
::
as
<
InferenceEngine
::
MemoryBlob
>
(
output_label
);
InferenceEngine
::
TensorDesc
blob_label
=
moutput_label
->
getTensorDesc
();
std
::
vector
<
size_t
>
output_label_shape
=
blob_label
.
getDims
();
size
=
1
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录