Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Greenplum
Opencv
提交
bc210b29
O
Opencv
项目概览
Greenplum
/
Opencv
10 个月 前同步成功
通知
7
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
O
Opencv
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
bc210b29
编写于
7月 09, 2021
作者:
A
Alexander Alekhin
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
dnn(test): backport test_ie_models.cpp from 4.5.3
上级
167a1202
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
22 addition
and
6 deletion
+22
-6
modules/dnn/test/test_ie_models.cpp
modules/dnn/test/test_ie_models.cpp
+22
-6
未找到文件。
modules/dnn/test/test_ie_models.cpp
浏览文件 @
bc210b29
...
...
@@ -103,11 +103,15 @@ static const std::map<std::string, OpenVINOModelTestCaseInfo>& getOpenVINOTestMo
#if INF_ENGINE_RELEASE >= 2020010000
// Downloaded using these parameters for Open Model Zoo downloader (2020.1):
// ./downloader.py -o ${OPENCV_DNN_TEST_DATA_PATH}/omz_intel_models --cache_dir ${OPENCV_DNN_TEST_DATA_PATH}/.omz_cache/ \
// --name person-detection-retail-0013
// --name person-detection-retail-0013
,age-gender-recognition-retail-0013
{
"person-detection-retail-0013"
,
{
// IRv10
"intel/person-detection-retail-0013/FP32/person-detection-retail-0013"
,
"intel/person-detection-retail-0013/FP16/person-detection-retail-0013"
}},
{
"age-gender-recognition-retail-0013"
,
{
"intel/age-gender-recognition-retail-0013/FP16/age-gender-recognition-retail-0013"
,
"intel/age-gender-recognition-retail-0013/FP32/age-gender-recognition-retail-0013"
}},
#endif
};
...
...
@@ -123,6 +127,21 @@ static const std::vector<std::string> getOpenVINOTestModelsList()
return
result
;
}
inline
static
std
::
string
getOpenVINOModel
(
const
std
::
string
&
modelName
,
bool
isFP16
)
{
const
std
::
map
<
std
::
string
,
OpenVINOModelTestCaseInfo
>&
models
=
getOpenVINOTestModels
();
const
auto
it
=
models
.
find
(
modelName
);
if
(
it
!=
models
.
end
())
{
OpenVINOModelTestCaseInfo
modelInfo
=
it
->
second
;
if
(
isFP16
&&
modelInfo
.
modelPathFP16
)
return
std
::
string
(
modelInfo
.
modelPathFP16
);
else
if
(
!
isFP16
&&
modelInfo
.
modelPathFP32
)
return
std
::
string
(
modelInfo
.
modelPathFP32
);
}
return
std
::
string
();
}
static
inline
void
genData
(
const
InferenceEngine
::
TensorDesc
&
desc
,
Mat
&
m
,
Blob
::
Ptr
&
dataPtr
)
{
const
std
::
vector
<
size_t
>&
dims
=
desc
.
getDims
();
...
...
@@ -319,11 +338,8 @@ TEST_P(DNNTestOpenVINO, models)
bool
isFP16
=
(
targetId
==
DNN_TARGET_OPENCL_FP16
||
targetId
==
DNN_TARGET_MYRIAD
);
const
std
::
map
<
std
::
string
,
OpenVINOModelTestCaseInfo
>&
models
=
getOpenVINOTestModels
();
const
auto
it
=
models
.
find
(
modelName
);
ASSERT_TRUE
(
it
!=
models
.
end
())
<<
modelName
;
OpenVINOModelTestCaseInfo
modelInfo
=
it
->
second
;
std
::
string
modelPath
=
isFP16
?
modelInfo
.
modelPathFP16
:
modelInfo
.
modelPathFP32
;
const
std
::
string
modelPath
=
getOpenVINOModel
(
modelName
,
isFP16
);
ASSERT_FALSE
(
modelPath
.
empty
())
<<
modelName
;
std
::
string
xmlPath
=
findDataFile
(
modelPath
+
".xml"
,
false
);
std
::
string
binPath
=
findDataFile
(
modelPath
+
".bin"
,
false
);
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录