Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
Paddle
提交
b6267919
P
Paddle
项目概览
PaddlePaddle
/
Paddle
11 个月 前同步成功
通知
2292
Star
20931
Fork
5422
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1423
列表
看板
标记
里程碑
合并请求
543
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1,423
Issue
1,423
列表
看板
标记
里程碑
合并请求
543
合并请求
543
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
b6267919
编写于
6月 01, 2021
作者:
W
Wangzheee
提交者:
GitHub
6月 01, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
[cmake] download_verify (#33217)
* download_verify * checksum
上级
519cc7b0
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
132 addition
and
69 deletion
+132
-69
paddle/fluid/inference/tests/api/CMakeLists.txt
paddle/fluid/inference/tests/api/CMakeLists.txt
+106
-66
paddle/fluid/inference/tests/test.cmake
paddle/fluid/inference/tests/test.cmake
+26
-3
未找到文件。
paddle/fluid/inference/tests/api/CMakeLists.txt
浏览文件 @
b6267919
...
@@ -8,44 +8,84 @@ if(WITH_GPU AND TENSORRT_FOUND)
...
@@ -8,44 +8,84 @@ if(WITH_GPU AND TENSORRT_FOUND)
set
(
INFERENCE_EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
analysis
${
analysis_deps
}
)
set
(
INFERENCE_EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
analysis
${
analysis_deps
}
)
endif
()
endif
()
function
(
download_data install_dir data_file
)
function
(
download_data install_dir data_file
check_sum
)
string
(
REGEX MATCH
"[^/
\\
]+$"
file_name
${
data_file
}
)
string
(
REGEX MATCH
"[^/
\\
]+$"
file_name
${
data_file
}
)
if
(
NOT EXISTS
${
install_dir
}
/
${
file_name
}
)
if
(
NOT EXISTS
${
install_dir
}
/
${
file_name
}
)
inference_download_and_uncompress
(
${
install_dir
}
${
INFERENCE_URL
}
${
data_file
}
)
inference_download_and_uncompress
(
${
install_dir
}
${
INFERENCE_URL
}
${
data_file
}
${
check_sum
}
)
endif
()
endif
()
endfunction
()
endfunction
()
function
(
download_int8_data install_dir data_file
)
function
(
download_data_without_verify install_dir data_file
)
string
(
REGEX MATCH
"[^/
\\
]+$"
file_name
${
data_file
}
)
if
(
NOT EXISTS
${
install_dir
}
/
${
file_name
}
)
inference_download_and_uncompress_without_verify
(
${
install_dir
}
${
INFERENCE_URL
}
${
data_file
}
)
endif
()
endfunction
()
function
(
download_int8_data install_dir data_file check_sum
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
inference_download_and_uncompress
(
${
install_dir
}
${
INFERENCE_URL
}
/int8
${
data_file
}
${
check_sum
}
)
endif
()
endfunction
()
function
(
download_int8_data_without_verify install_dir data_file
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
inference_download_and_uncompress_without_verify
(
${
install_dir
}
${
INFERENCE_URL
}
/int8
${
data_file
}
)
endif
()
endfunction
()
function
(
download_bfloat16_data install_dir data_file check_sum
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
inference_download_and_uncompress
(
${
install_dir
}
${
INFERENCE_URL
}
/
int8
${
data_file
}
)
inference_download_and_uncompress
(
${
install_dir
}
${
INFERENCE_URL
}
/
bfloat16
${
data_file
}
${
check_sum
}
)
endif
()
endif
()
endfunction
()
endfunction
()
function
(
download_bfloat16_data install_dir data_file
)
function
(
download_bfloat16_data
_without_verify
install_dir data_file
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
inference_download_and_uncompress
(
${
install_dir
}
${
INFERENCE_URL
}
/bfloat16
${
data_file
}
)
inference_download_and_uncompress
_without_verify
(
${
install_dir
}
${
INFERENCE_URL
}
/bfloat16
${
data_file
}
)
endif
()
endif
()
endfunction
()
endfunction
()
function
(
download_GRU_data install_dir data_file
)
function
(
download_GRU_data install_dir data_file
check_sum
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
inference_download_and_uncompress
(
${
install_dir
}
${
INFERENCE_URL
}
/gru
${
data_file
}
)
inference_download_and_uncompress
(
${
install_dir
}
${
INFERENCE_URL
}
/gru
${
data_file
}
${
check_sum
}
)
endif
()
endif
()
endfunction
()
endfunction
()
function
(
download_
quant_data
install_dir data_file
)
function
(
download_
GRU_data_without_verify
install_dir data_file
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
inference_download_and_uncompress
(
${
install_dir
}
${
INFERENCE_URL
}
/int8/QAT_models
${
data_file
}
)
inference_download_and_uncompress_without_verify
(
${
install_dir
}
${
INFERENCE_URL
}
/gru
${
data_file
}
)
endif
()
endif
()
endfunction
()
endfunction
()
function
(
download_model_and_data install_dir model_name data_name
)
function
(
download_quant_data install_dir data_file check_sum
)
download_data
(
${
install_dir
}
${
model_name
}
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
download_data
(
${
install_dir
}
${
data_name
}
)
inference_download_and_uncompress
(
${
install_dir
}
${
INFERENCE_URL
}
/int8/QAT_models
${
data_file
}
${
check_sum
}
)
endif
()
endfunction
()
function
(
download_quant_data_without_verify install_dir data_file
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
inference_download_and_uncompress_without_verify
(
${
install_dir
}
${
INFERENCE_URL
}
/int8/QAT_models
${
data_file
}
)
endif
()
endfunction
()
function
(
download_model_and_data install_dir model_name model_check_sum data_name data_check_sum
)
download_data
(
${
install_dir
}
${
model_name
}
${
model_check_sum
}
)
download_data
(
${
install_dir
}
${
data_name
}
${
data_check_sum
}
)
endfunction
()
function
(
download_model_and_data_without_verify install_dir model_name data_name
)
download_data_without_verify
(
${
install_dir
}
${
model_name
}
)
download_data_without_verify
(
${
install_dir
}
${
data_name
}
)
endfunction
()
function
(
download_result install_dir result_name check_sum
)
download_data
(
${
install_dir
}
${
result_name
}
${
check_sum
}
)
endfunction
()
endfunction
()
function
(
download_result install_dir result_name
)
function
(
download_result
_without_verify
install_dir result_name
)
download_data
(
${
install_dir
}
${
result_name
}
)
download_data
_without_verify
(
${
install_dir
}
${
result_name
}
)
endfunction
()
endfunction
()
function
(
inference_analysis_api_test target install_dir filename
)
function
(
inference_analysis_api_test target install_dir filename
)
...
@@ -165,12 +205,12 @@ endfunction()
...
@@ -165,12 +205,12 @@ endfunction()
if
(
NOT APPLE AND WITH_MKLML
)
if
(
NOT APPLE AND WITH_MKLML
)
# RNN1
# RNN1
set
(
RNN1_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/rnn1"
)
set
(
RNN1_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/rnn1"
)
download_model_and_data
(
${
RNN1_INSTALL_DIR
}
"rnn1/model.tar.gz"
"rnn1/data.txt.tar.gz"
)
download_model_and_data
_without_verify
(
${
RNN1_INSTALL_DIR
}
"rnn1/model.tar.gz"
"rnn1/data.txt.tar.gz"
)
inference_analysis_api_test
(
test_analyzer_rnn1
${
RNN1_INSTALL_DIR
}
analyzer_rnn1_tester.cc
)
inference_analysis_api_test
(
test_analyzer_rnn1
${
RNN1_INSTALL_DIR
}
analyzer_rnn1_tester.cc
)
# seq_pool1
# seq_pool1
set
(
SEQ_POOL1_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/seq_pool"
)
set
(
SEQ_POOL1_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/seq_pool"
)
download_model_and_data
(
${
SEQ_POOL1_INSTALL_DIR
}
"seq_pool1_model_.tar.gz"
"seq_pool1_data.txt.tar.gz"
)
download_model_and_data
_without_verify
(
${
SEQ_POOL1_INSTALL_DIR
}
"seq_pool1_model_.tar.gz"
"seq_pool1_data.txt.tar.gz"
)
inference_analysis_api_test
(
test_analyzer_seq_pool1_compare_determine
${
SEQ_POOL1_INSTALL_DIR
}
analyzer_seq_pool1_compare_determine_tester.cc
)
inference_analysis_api_test
(
test_analyzer_seq_pool1_compare_determine
${
SEQ_POOL1_INSTALL_DIR
}
analyzer_seq_pool1_compare_determine_tester.cc
)
inference_analysis_api_test
(
test_analyzer_seq_pool1
${
SEQ_POOL1_INSTALL_DIR
}
analyzer_seq_pool1_compare_tester.cc
)
inference_analysis_api_test
(
test_analyzer_seq_pool1
${
SEQ_POOL1_INSTALL_DIR
}
analyzer_seq_pool1_compare_tester.cc
)
inference_analysis_api_test
(
test_analyzer_seq_pool1_fuse_compare_zero_copy
${
SEQ_POOL1_INSTALL_DIR
}
analyzer_seq_pool1_fuse_compare_zero_copy_tester.cc
)
inference_analysis_api_test
(
test_analyzer_seq_pool1_fuse_compare_zero_copy
${
SEQ_POOL1_INSTALL_DIR
}
analyzer_seq_pool1_fuse_compare_zero_copy_tester.cc
)
...
@@ -193,7 +233,7 @@ endif()
...
@@ -193,7 +233,7 @@ endif()
# RNN2
# RNN2
set
(
RNN2_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/rnn2"
)
set
(
RNN2_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/rnn2"
)
download_model_and_data
(
${
RNN2_INSTALL_DIR
}
"rnn2_model.tar.gz"
"rnn2_data.txt.tar.gz"
)
download_model_and_data
_without_verify
(
${
RNN2_INSTALL_DIR
}
"rnn2_model.tar.gz"
"rnn2_data.txt.tar.gz"
)
inference_analysis_api_test
(
test_analyzer_rnn2
${
RNN2_INSTALL_DIR
}
analyzer_rnn2_tester.cc
)
inference_analysis_api_test
(
test_analyzer_rnn2
${
RNN2_INSTALL_DIR
}
analyzer_rnn2_tester.cc
)
# TODO(luotao, Superjom) Disable DAM test, temporarily fix
# TODO(luotao, Superjom) Disable DAM test, temporarily fix
...
@@ -201,12 +241,12 @@ inference_analysis_api_test(test_analyzer_rnn2 ${RNN2_INSTALL_DIR} analyzer_rnn2
...
@@ -201,12 +241,12 @@ inference_analysis_api_test(test_analyzer_rnn2 ${RNN2_INSTALL_DIR} analyzer_rnn2
# After inference framework refactor, will reopen it.
# After inference framework refactor, will reopen it.
# normal DAM
# normal DAM
set
(
DAM_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/dam"
)
set
(
DAM_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/dam"
)
download_model_and_data
(
${
DAM_INSTALL_DIR
}
"DAM_model.tar.gz"
"DAM_data.txt.tar.gz"
)
download_model_and_data
_without_verify
(
${
DAM_INSTALL_DIR
}
"DAM_model.tar.gz"
"DAM_data.txt.tar.gz"
)
#inference_analysis_api_test(test_analyzer_dam ${DAM_INSTALL_DIR} analyzer_dam_tester.cc EXTRA_DEPS legacy_allocator)
#inference_analysis_api_test(test_analyzer_dam ${DAM_INSTALL_DIR} analyzer_dam_tester.cc EXTRA_DEPS legacy_allocator)
# small DAM
# small DAM
set
(
DAM_SMALL_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/small_dam"
)
set
(
DAM_SMALL_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/small_dam"
)
download_model_and_data
(
${
DAM_SMALL_INSTALL_DIR
}
"dam_small_model.tar.gz"
"dam_small_data.txt.tar.gz"
)
download_model_and_data
_without_verify
(
${
DAM_SMALL_INSTALL_DIR
}
"dam_small_model.tar.gz"
"dam_small_data.txt.tar.gz"
)
inference_analysis_test
(
test_analyzer_small_dam SRCS analyzer_dam_tester.cc
inference_analysis_test
(
test_analyzer_small_dam SRCS analyzer_dam_tester.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
ARGS --infer_model=
${
DAM_SMALL_INSTALL_DIR
}
/model --infer_data=
${
DAM_SMALL_INSTALL_DIR
}
/data.txt
)
ARGS --infer_model=
${
DAM_SMALL_INSTALL_DIR
}
/model --infer_data=
${
DAM_SMALL_INSTALL_DIR
}
/data.txt
)
...
@@ -216,29 +256,29 @@ inference_analysis_api_test(test_analyzer_save_model ${DAM_SMALL_INSTALL_DIR} an
...
@@ -216,29 +256,29 @@ inference_analysis_api_test(test_analyzer_save_model ${DAM_SMALL_INSTALL_DIR} an
# chinese_ner
# chinese_ner
set
(
CHINESE_NER_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/chinese_ner"
)
set
(
CHINESE_NER_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/chinese_ner"
)
download_model_and_data
(
${
CHINESE_NER_INSTALL_DIR
}
"chinese_ner_model.tar.gz"
"chinese_ner-data.txt.tar.gz"
)
download_model_and_data
_without_verify
(
${
CHINESE_NER_INSTALL_DIR
}
"chinese_ner_model.tar.gz"
"chinese_ner-data.txt.tar.gz"
)
inference_analysis_api_test
(
test_analyzer_ner
${
CHINESE_NER_INSTALL_DIR
}
analyzer_ner_tester.cc
)
inference_analysis_api_test
(
test_analyzer_ner
${
CHINESE_NER_INSTALL_DIR
}
analyzer_ner_tester.cc
)
# lac
# lac
set
(
LAC_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/lac"
)
set
(
LAC_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/lac"
)
download_model_and_data
(
${
LAC_INSTALL_DIR
}
"lac_model.tar.gz"
"lac_data.txt.tar.gz"
)
download_model_and_data
(
${
LAC_INSTALL_DIR
}
"lac_model.tar.gz"
419ca6eb85f57a01bfe173591910aec5
"lac_data.txt.tar.gz"
9983539cd6b34fbdc411e43422776bfd
)
inference_analysis_api_test
(
test_analyzer_lac
${
LAC_INSTALL_DIR
}
analyzer_lac_tester.cc
)
inference_analysis_api_test
(
test_analyzer_lac
${
LAC_INSTALL_DIR
}
analyzer_lac_tester.cc
)
# Pyramid DNN
# Pyramid DNN
set
(
PYRAMID_DNN_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/pyramid_dnn"
)
set
(
PYRAMID_DNN_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/pyramid_dnn"
)
download_model_and_data
(
${
PYRAMID_DNN_INSTALL_DIR
}
"PyramidDNN_model.tar.gz"
"PyramidDNN_data.txt.tar.gz"
)
download_model_and_data
_without_verify
(
${
PYRAMID_DNN_INSTALL_DIR
}
"PyramidDNN_model.tar.gz"
"PyramidDNN_data.txt.tar.gz"
)
inference_analysis_api_test
(
test_analyzer_pyramid_dnn
${
PYRAMID_DNN_INSTALL_DIR
}
analyzer_pyramid_dnn_tester.cc
)
inference_analysis_api_test
(
test_analyzer_pyramid_dnn
${
PYRAMID_DNN_INSTALL_DIR
}
analyzer_pyramid_dnn_tester.cc
)
#Ernie
#Ernie
set
(
ERNIE_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/Ernie"
)
set
(
ERNIE_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/Ernie"
)
download_model_and_data
(
${
ERNIE_INSTALL_DIR
}
"Ernie_model.tar.gz"
"Ernie_data.txt.tar.gz"
"Ernie_result.txt.tar.gz"
)
download_model_and_data
(
${
ERNIE_INSTALL_DIR
}
"Ernie_model.tar.gz"
aa59192dd41ed377f9f168e3a1309fa6
"Ernie_data.txt.tar.gz"
5396e63548edad7ca561e7e26a9476d1
)
download_result
(
${
ERNIE_INSTALL_DIR
}
"Ernie_result.txt.tar.gz"
)
download_result
(
${
ERNIE_INSTALL_DIR
}
"Ernie_result.txt.tar.gz"
73beea65abda2edb61c1662cd3180c62
)
inference_analysis_api_test
(
test_analyzer_ernie
${
ERNIE_INSTALL_DIR
}
analyzer_ernie_tester.cc
)
inference_analysis_api_test
(
test_analyzer_ernie
${
ERNIE_INSTALL_DIR
}
analyzer_ernie_tester.cc
)
#Ernie large
#Ernie large
set
(
ERNIE_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/Ernie_Large"
)
set
(
ERNIE_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/Ernie_Large"
)
download_model_and_data
(
${
ERNIE_INSTALL_DIR
}
"Ernie_large_model.tar.gz"
"Ernie_large_data.txt.tar.gz"
"Ernie_large_result.txt.tar.gz"
)
download_model_and_data
(
${
ERNIE_INSTALL_DIR
}
"Ernie_large_model.tar.gz"
af7715245ed32cc77374625d4c80f7ef
"Ernie_large_data.txt.tar.gz"
edb2113eec93783cad56ed76d47ba57f
)
download_result
(
${
ERNIE_INSTALL_DIR
}
"Ernie_large_result.txt.tar.gz"
)
download_result
(
${
ERNIE_INSTALL_DIR
}
"Ernie_large_result.txt.tar.gz"
1facda98eef1085dc9d435ebf3f23a73
)
inference_analysis_test
(
test_analyzer_ernie_large SRCS analyzer_ernie_tester.cc
inference_analysis_test
(
test_analyzer_ernie_large SRCS analyzer_ernie_tester.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
ARGS --infer_model=
${
ERNIE_INSTALL_DIR
}
/model --infer_data=
${
ERNIE_INSTALL_DIR
}
/data.txt --refer_result=
${
ERNIE_INSTALL_DIR
}
/result.txt --ernie_large=true
)
ARGS --infer_model=
${
ERNIE_INSTALL_DIR
}
/model --infer_data=
${
ERNIE_INSTALL_DIR
}
/data.txt --refer_result=
${
ERNIE_INSTALL_DIR
}
/result.txt --ernie_large=true
)
...
@@ -251,17 +291,17 @@ endif()
...
@@ -251,17 +291,17 @@ endif()
# text_classification
# text_classification
set
(
TEXT_CLASSIFICATION_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/text_classification"
)
set
(
TEXT_CLASSIFICATION_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/text_classification"
)
download_model_and_data
(
${
TEXT_CLASSIFICATION_INSTALL_DIR
}
"text-classification-Senta.tar.gz"
"text_classification_data.txt.tar.gz"
)
download_model_and_data
(
${
TEXT_CLASSIFICATION_INSTALL_DIR
}
"text-classification-Senta.tar.gz"
3f0f440313ca50e26184e65ffd5809ab
"text_classification_data.txt.tar.gz"
36ae620020cc3377f45ed330dd36238f
)
inference_analysis_api_test
(
test_analyzer_text_classification
${
TEXT_CLASSIFICATION_INSTALL_DIR
}
analyzer_text_classification_tester.cc
)
inference_analysis_api_test
(
test_analyzer_text_classification
${
TEXT_CLASSIFICATION_INSTALL_DIR
}
analyzer_text_classification_tester.cc
)
# seq_conv1
# seq_conv1
set
(
SEQ_CONV1_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/seq_conv1"
)
set
(
SEQ_CONV1_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/seq_conv1"
)
download_model_and_data
(
${
SEQ_CONV1_INSTALL_DIR
}
"seq_conv1_model.tar.gz"
"seq_conv1_data.txt.tar.gz"
)
download_model_and_data
_without_verify
(
${
SEQ_CONV1_INSTALL_DIR
}
"seq_conv1_model.tar.gz"
"seq_conv1_data.txt.tar.gz"
)
inference_analysis_api_test
(
test_analyzer_seq_conv1
${
SEQ_CONV1_INSTALL_DIR
}
analyzer_seq_conv1_tester.cc
)
inference_analysis_api_test
(
test_analyzer_seq_conv1
${
SEQ_CONV1_INSTALL_DIR
}
analyzer_seq_conv1_tester.cc
)
# transformer, the dataset only works on batch_size=8 now
# transformer, the dataset only works on batch_size=8 now
set
(
TRANSFORMER_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/transformer"
)
set
(
TRANSFORMER_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/transformer"
)
download_model_and_data
(
${
TRANSFORMER_INSTALL_DIR
}
"temp/transformer_model.tar.gz"
"temp/transformer_data.txt.tar.gz"
)
download_model_and_data
_without_verify
(
${
TRANSFORMER_INSTALL_DIR
}
"temp/transformer_model.tar.gz"
"temp/transformer_data.txt.tar.gz"
)
inference_analysis_test
(
test_analyzer_transformer SRCS analyzer_transformer_compare_tester.cc
inference_analysis_test
(
test_analyzer_transformer SRCS analyzer_transformer_compare_tester.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
ARGS --infer_model=
${
TRANSFORMER_INSTALL_DIR
}
/model --infer_data=
${
TRANSFORMER_INSTALL_DIR
}
/data.txt --batch_size=8
ARGS --infer_model=
${
TRANSFORMER_INSTALL_DIR
}
/model --infer_data=
${
TRANSFORMER_INSTALL_DIR
}
/data.txt --batch_size=8
...
@@ -278,13 +318,13 @@ inference_analysis_test(test_analyzer_transformer_profile SRCS analyzer_transfor
...
@@ -278,13 +318,13 @@ inference_analysis_test(test_analyzer_transformer_profile SRCS analyzer_transfor
# ocr
# ocr
set
(
OCR_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/ocr"
)
set
(
OCR_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/ocr"
)
if
(
NOT EXISTS
${
OCR_INSTALL_DIR
}
/ocr.tar.gz
)
if
(
NOT EXISTS
${
OCR_INSTALL_DIR
}
/ocr.tar.gz
)
inference_download_and_uncompress
(
${
OCR_INSTALL_DIR
}
"http://paddlemodels.bj.bcebos.com/"
"inference-vis-demos/ocr.tar.gz"
)
inference_download_and_uncompress
_without_verify
(
${
OCR_INSTALL_DIR
}
"http://paddlemodels.bj.bcebos.com/"
"inference-vis-demos/ocr.tar.gz"
)
endif
()
endif
()
inference_analysis_api_test
(
test_analyzer_ocr
${
OCR_INSTALL_DIR
}
analyzer_vis_tester.cc
)
inference_analysis_api_test
(
test_analyzer_ocr
${
OCR_INSTALL_DIR
}
analyzer_vis_tester.cc
)
# densebox
# densebox
set
(
DENSEBOX_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/densebox"
)
set
(
DENSEBOX_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/densebox"
)
download_data
(
${
DENSEBOX_INSTALL_DIR
}
"densebox.tar.gz"
)
download_data
_without_verify
(
${
DENSEBOX_INSTALL_DIR
}
"densebox.tar.gz"
)
#inference_analysis_test(test_analyzer_detect SRCS analyzer_detect_tester.cc
#inference_analysis_test(test_analyzer_detect SRCS analyzer_detect_tester.cc
# EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
# EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
# ARGS --infer_model=${DENSEBOX_INSTALL_DIR}/model --infer_data=${DENSEBOX_INSTALL_DIR}/detect_input_50.txt
# ARGS --infer_model=${DENSEBOX_INSTALL_DIR}/model --infer_data=${DENSEBOX_INSTALL_DIR}/detect_input_50.txt
...
@@ -294,7 +334,7 @@ download_data(${DENSEBOX_INSTALL_DIR} "densebox.tar.gz")
...
@@ -294,7 +334,7 @@ download_data(${DENSEBOX_INSTALL_DIR} "densebox.tar.gz")
# mobilenet with transpose op
# mobilenet with transpose op
set
(
MOBILENET_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/mobilenet"
)
set
(
MOBILENET_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/mobilenet"
)
if
(
NOT EXISTS
${
MOBILENET_INSTALL_DIR
}
/mobilenet.tar.gz
)
if
(
NOT EXISTS
${
MOBILENET_INSTALL_DIR
}
/mobilenet.tar.gz
)
inference_download_and_uncompress
(
${
MOBILENET_INSTALL_DIR
}
"http://paddlemodels.bj.bcebos.com/"
"inference-vis-demos/mobilenet.tar.gz"
)
inference_download_and_uncompress
_without_verify
(
${
MOBILENET_INSTALL_DIR
}
"http://paddlemodels.bj.bcebos.com/"
"inference-vis-demos/mobilenet.tar.gz"
)
endif
()
endif
()
inference_analysis_api_test
(
test_analyzer_mobilenet_transpose
${
MOBILENET_INSTALL_DIR
}
analyzer_vis_tester.cc
)
inference_analysis_api_test
(
test_analyzer_mobilenet_transpose
${
MOBILENET_INSTALL_DIR
}
analyzer_vis_tester.cc
)
...
@@ -307,13 +347,13 @@ inference_analysis_api_test_with_fake_data_build(${IMG_CLASS_TEST_APP} ${IMG_CLA
...
@@ -307,13 +347,13 @@ inference_analysis_api_test_with_fake_data_build(${IMG_CLASS_TEST_APP} ${IMG_CLA
# googlenet
# googlenet
set
(
GOOGLENET_MODEL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/googlenet"
)
set
(
GOOGLENET_MODEL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/googlenet"
)
download_data
(
${
GOOGLENET_MODEL_DIR
}
"googlenet.tar.gz"
)
download_data
_without_verify
(
${
GOOGLENET_MODEL_DIR
}
"googlenet.tar.gz"
)
inference_analysis_api_test_with_fake_data_run
(
test_analyzer_googlenet
${
IMG_CLASS_TEST_APP
}
inference_analysis_api_test_with_fake_data_run
(
test_analyzer_googlenet
${
IMG_CLASS_TEST_APP
}
${
GOOGLENET_MODEL_DIR
}
false
)
${
GOOGLENET_MODEL_DIR
}
false
)
# resnet50
# resnet50
set
(
RESNET50_MODEL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/resnet50"
)
set
(
RESNET50_MODEL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/resnet50"
)
download_data
(
${
RESNET50_MODEL_DIR
}
"resnet50_model.tar.gz"
)
download_data
_without_verify
(
${
RESNET50_MODEL_DIR
}
"resnet50_model.tar.gz"
)
inference_analysis_api_test_with_fake_data_run
(
test_analyzer_resnet50
${
IMG_CLASS_TEST_APP
}
inference_analysis_api_test_with_fake_data_run
(
test_analyzer_resnet50
${
IMG_CLASS_TEST_APP
}
${
RESNET50_MODEL_DIR
}
true
)
${
RESNET50_MODEL_DIR
}
true
)
if
(
WIN32
)
if
(
WIN32
)
...
@@ -323,7 +363,7 @@ endif()
...
@@ -323,7 +363,7 @@ endif()
# mobilenet with depthwise_conv op
# mobilenet with depthwise_conv op
set
(
MOBILENET_MODEL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/mobilenet_depthwise_conv"
)
set
(
MOBILENET_MODEL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/mobilenet_depthwise_conv"
)
download_data
(
${
MOBILENET_MODEL_DIR
}
"mobilenet_model.tar.gz"
)
download_data
_without_verify
(
${
MOBILENET_MODEL_DIR
}
"mobilenet_model.tar.gz"
)
inference_analysis_api_test_with_fake_data_run
(
test_analyzer_mobilenet_depthwise_conv
${
IMG_CLASS_TEST_APP
}
inference_analysis_api_test_with_fake_data_run
(
test_analyzer_mobilenet_depthwise_conv
${
IMG_CLASS_TEST_APP
}
${
MOBILENET_MODEL_DIR
}
false
)
${
MOBILENET_MODEL_DIR
}
false
)
...
@@ -340,7 +380,7 @@ if(WITH_MKLDNN)
...
@@ -340,7 +380,7 @@ if(WITH_MKLDNN)
set
(
IMAGENET_DATA_ARCHIVE
"imagenet_val_100_tail.tar.gz"
)
set
(
IMAGENET_DATA_ARCHIVE
"imagenet_val_100_tail.tar.gz"
)
set
(
IMAGENET_DATA_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/imagenet"
)
set
(
IMAGENET_DATA_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/imagenet"
)
set
(
IMAGENET_DATA_PATH
"
${
IMAGENET_DATA_DIR
}
/data.bin"
)
set
(
IMAGENET_DATA_PATH
"
${
IMAGENET_DATA_DIR
}
/data.bin"
)
download_int8_data
(
${
IMAGENET_DATA_DIR
}
${
IMAGENET_DATA_ARCHIVE
}
)
download_int8_data
_without_verify
(
${
IMAGENET_DATA_DIR
}
${
IMAGENET_DATA_ARCHIVE
}
)
# build test binary to be used in subsequent tests
# build test binary to be used in subsequent tests
set
(
INT8_IMG_CLASS_TEST_APP
"test_analyzer_int8_image_classification"
)
set
(
INT8_IMG_CLASS_TEST_APP
"test_analyzer_int8_image_classification"
)
...
@@ -349,40 +389,40 @@ if(WITH_MKLDNN)
...
@@ -349,40 +389,40 @@ if(WITH_MKLDNN)
# resnet50 int8
# resnet50 int8
set
(
INT8_RESNET50_MODEL_DIR
"
${
INT8_DATA_DIR
}
/resnet50"
)
set
(
INT8_RESNET50_MODEL_DIR
"
${
INT8_DATA_DIR
}
/resnet50"
)
download_int8_data
(
${
INT8_RESNET50_MODEL_DIR
}
"resnet50_int8_model.tar.gz"
)
download_int8_data
_without_verify
(
${
INT8_RESNET50_MODEL_DIR
}
"resnet50_int8_model.tar.gz"
)
inference_analysis_api_int8_test_run
(
test_analyzer_int8_resnet50
${
INT8_IMG_CLASS_TEST_APP
}
${
INT8_RESNET50_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
)
inference_analysis_api_int8_test_run
(
test_analyzer_int8_resnet50
${
INT8_IMG_CLASS_TEST_APP
}
${
INT8_RESNET50_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
)
# mobilenetv1 int8
# mobilenetv1 int8
set
(
INT8_MOBILENETV1_MODEL_DIR
"
${
INT8_DATA_DIR
}
/mobilenetv1"
)
set
(
INT8_MOBILENETV1_MODEL_DIR
"
${
INT8_DATA_DIR
}
/mobilenetv1"
)
download_int8_data
(
${
INT8_MOBILENETV1_MODEL_DIR
}
"mobilenetv1_int8_model.tar.gz"
)
download_int8_data
_without_verify
(
${
INT8_MOBILENETV1_MODEL_DIR
}
"mobilenetv1_int8_model.tar.gz"
)
inference_analysis_api_int8_test_run
(
test_analyzer_int8_mobilenetv1
${
INT8_IMG_CLASS_TEST_APP
}
${
INT8_MOBILENETV1_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
)
inference_analysis_api_int8_test_run
(
test_analyzer_int8_mobilenetv1
${
INT8_IMG_CLASS_TEST_APP
}
${
INT8_MOBILENETV1_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
)
# mobilenetv2 int8
# mobilenetv2 int8
set
(
INT8_MOBILENETV2_MODEL_DIR
"
${
INT8_DATA_DIR
}
/mobilenetv2"
)
set
(
INT8_MOBILENETV2_MODEL_DIR
"
${
INT8_DATA_DIR
}
/mobilenetv2"
)
download_int8_data
(
${
INT8_MOBILENETV2_MODEL_DIR
}
"mobilenet_v2_int8_model.tar.gz"
)
download_int8_data
_without_verify
(
${
INT8_MOBILENETV2_MODEL_DIR
}
"mobilenet_v2_int8_model.tar.gz"
)
inference_analysis_api_int8_test_run
(
test_analyzer_int8_mobilenetv2
${
INT8_IMG_CLASS_TEST_APP
}
${
INT8_MOBILENETV2_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
)
inference_analysis_api_int8_test_run
(
test_analyzer_int8_mobilenetv2
${
INT8_IMG_CLASS_TEST_APP
}
${
INT8_MOBILENETV2_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
)
# resnet101 int8
# resnet101 int8
# TODO(grygielski) Enable after MKL-DNN 1.0 merge
# TODO(grygielski) Enable after MKL-DNN 1.0 merge
set
(
INT8_RESNET101_MODEL_DIR
"
${
INT8_DATA_DIR
}
/resnet101"
)
set
(
INT8_RESNET101_MODEL_DIR
"
${
INT8_DATA_DIR
}
/resnet101"
)
download_int8_data
(
${
INT8_RESNET101_MODEL_DIR
}
"Res101_int8_model.tar.gz"
)
download_int8_data
_without_verify
(
${
INT8_RESNET101_MODEL_DIR
}
"Res101_int8_model.tar.gz"
)
# inference_analysis_api_int8_test_run(test_analyzer_int8_resnet101 ${INT8_IMG_CLASS_TEST_APP} ${INT8_RESNET101_MODEL_DIR} ${IMAGENET_DATA_PATH})
# inference_analysis_api_int8_test_run(test_analyzer_int8_resnet101 ${INT8_IMG_CLASS_TEST_APP} ${INT8_RESNET101_MODEL_DIR} ${IMAGENET_DATA_PATH})
# vgg16 int8
# vgg16 int8
# TODO(grygielski) Enable after MKL-DNN 1.0 merge
# TODO(grygielski) Enable after MKL-DNN 1.0 merge
set
(
INT8_VGG16_MODEL_DIR
"
${
INT8_DATA_DIR
}
/vgg16"
)
set
(
INT8_VGG16_MODEL_DIR
"
${
INT8_DATA_DIR
}
/vgg16"
)
download_int8_data
(
${
INT8_VGG16_MODEL_DIR
}
"VGG16_int8_model.tar.gz"
)
download_int8_data
_without_verify
(
${
INT8_VGG16_MODEL_DIR
}
"VGG16_int8_model.tar.gz"
)
# inference_analysis_api_int8_test_run(test_analyzer_int8_vgg16 ${INT8_IMG_CLASS_TEST_APP} ${INT8_VGG16_MODEL_DIR} ${IMAGENET_DATA_PATH})
# inference_analysis_api_int8_test_run(test_analyzer_int8_vgg16 ${INT8_IMG_CLASS_TEST_APP} ${INT8_VGG16_MODEL_DIR} ${IMAGENET_DATA_PATH})
# vgg19 int8
# vgg19 int8
# TODO(grygielski) Enable after MKL-DNN 1.0 merge
# TODO(grygielski) Enable after MKL-DNN 1.0 merge
set
(
INT8_VGG19_MODEL_DIR
"
${
INT8_DATA_DIR
}
/vgg19"
)
set
(
INT8_VGG19_MODEL_DIR
"
${
INT8_DATA_DIR
}
/vgg19"
)
download_int8_data
(
${
INT8_VGG19_MODEL_DIR
}
"VGG19_int8_model.tar.gz"
)
download_int8_data
_without_verify
(
${
INT8_VGG19_MODEL_DIR
}
"VGG19_int8_model.tar.gz"
)
# inference_analysis_api_int8_test_run(test_analyzer_int8_vgg19 ${INT8_IMG_CLASS_TEST_APP} ${INT8_VGG19_MODEL_DIR} ${IMAGENET_DATA_PATH})
# inference_analysis_api_int8_test_run(test_analyzer_int8_vgg19 ${INT8_IMG_CLASS_TEST_APP} ${INT8_VGG19_MODEL_DIR} ${IMAGENET_DATA_PATH})
# googlenet int8
# googlenet int8
set
(
INT8_GOOGLENET_MODEL_DIR
"
${
INT8_DATA_DIR
}
/googlenet"
)
set
(
INT8_GOOGLENET_MODEL_DIR
"
${
INT8_DATA_DIR
}
/googlenet"
)
download_int8_data
(
${
INT8_GOOGLENET_MODEL_DIR
}
"GoogleNet_int8_model.tar.gz"
)
download_int8_data
_without_verify
(
${
INT8_GOOGLENET_MODEL_DIR
}
"GoogleNet_int8_model.tar.gz"
)
inference_analysis_api_int8_test_run_custom_warmup_batch_size
(
test_analyzer_int8_googlenet
${
INT8_IMG_CLASS_TEST_APP
}
${
INT8_GOOGLENET_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
10
)
inference_analysis_api_int8_test_run_custom_warmup_batch_size
(
test_analyzer_int8_googlenet
${
INT8_IMG_CLASS_TEST_APP
}
${
INT8_GOOGLENET_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
10
)
### BFLOAT16 tests
### BFLOAT16 tests
...
@@ -410,7 +450,7 @@ if(WITH_MKLDNN)
...
@@ -410,7 +450,7 @@ if(WITH_MKLDNN)
set
(
INT8_OBJ_DETECT_TEST_APP_SRC
"analyzer_int8_object_detection_tester.cc"
)
set
(
INT8_OBJ_DETECT_TEST_APP_SRC
"analyzer_int8_object_detection_tester.cc"
)
# download dataset if necessary
# download dataset if necessary
download_int8_data
(
${
INT8_DATA_DIR
}
"pascalvoc_val_head_300.tar.gz"
)
download_int8_data
_without_verify
(
${
INT8_DATA_DIR
}
"pascalvoc_val_head_300.tar.gz"
)
# build test binary to be used in subsequent tests
# build test binary to be used in subsequent tests
...
@@ -418,13 +458,13 @@ if(WITH_MKLDNN)
...
@@ -418,13 +458,13 @@ if(WITH_MKLDNN)
# mobilenet-ssd int8
# mobilenet-ssd int8
set
(
INT8_MOBILENET_SSD_MODEL_DIR
"
${
INT8_DATA_DIR
}
/mobilenet-ssd"
)
set
(
INT8_MOBILENET_SSD_MODEL_DIR
"
${
INT8_DATA_DIR
}
/mobilenet-ssd"
)
download_int8_data
(
${
INT8_MOBILENET_SSD_MODEL_DIR
}
"mobilenet_ssd_int8_model.tar.gz"
)
download_int8_data
_without_verify
(
${
INT8_MOBILENET_SSD_MODEL_DIR
}
"mobilenet_ssd_int8_model.tar.gz"
)
inference_analysis_api_object_dection_int8_test_run
(
test_analyzer_int8_mobilenet_ssd
${
INT8_OBJ_DETECT_TEST_APP
}
${
INT8_MOBILENET_SSD_MODEL_DIR
}
${
PASCALVOC_DATA_PATH
}
)
inference_analysis_api_object_dection_int8_test_run
(
test_analyzer_int8_mobilenet_ssd
${
INT8_OBJ_DETECT_TEST_APP
}
${
INT8_MOBILENET_SSD_MODEL_DIR
}
${
PASCALVOC_DATA_PATH
}
)
### Lexcial analysis GRU model
### Lexcial analysis GRU model
set
(
GRU_PATH
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/gru"
)
set
(
GRU_PATH
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/gru"
)
download_GRU_data
(
"
${
GRU_PATH
}
"
"GRU_eval_data.tar.gz"
)
download_GRU_data
_without_verify
(
"
${
GRU_PATH
}
"
"GRU_eval_data.tar.gz"
)
download_GRU_data
(
"
${
GRU_PATH
}
"
"GRU_eval_model_v2.tar.gz"
)
download_GRU_data
_without_verify
(
"
${
GRU_PATH
}
"
"GRU_eval_model_v2.tar.gz"
)
set
(
GRU_DATA_PATH
"
${
GRU_PATH
}
/GRU_eval_data.bin"
)
set
(
GRU_DATA_PATH
"
${
GRU_PATH
}
/GRU_eval_data.bin"
)
set
(
GRU_MODEL_PATH
"
${
GRU_PATH
}
/GRU_eval_model_v2"
)
set
(
GRU_MODEL_PATH
"
${
GRU_PATH
}
/GRU_eval_model_v2"
)
set
(
LEXICAL_TEST_APP
"test_analyzer_lexical_analysis"
)
set
(
LEXICAL_TEST_APP
"test_analyzer_lexical_analysis"
)
...
@@ -451,9 +491,9 @@ if(WITH_MKLDNN)
...
@@ -451,9 +491,9 @@ if(WITH_MKLDNN)
set
(
QUANT2_MobileNetV1_MODEL_DIR
"
${
QUANT_DATA_DIR
}
/MobileNetV1_quant2"
)
set
(
QUANT2_MobileNetV1_MODEL_DIR
"
${
QUANT_DATA_DIR
}
/MobileNetV1_quant2"
)
set
(
QUANT2_INT8_MobileNetV1_MODEL_DIR
"
${
QUANT_DATA_DIR
}
/MobileNetV1_quant2_int8"
)
set
(
QUANT2_INT8_MobileNetV1_MODEL_DIR
"
${
QUANT_DATA_DIR
}
/MobileNetV1_quant2_int8"
)
if
(
NOT LINUX
)
if
(
NOT LINUX
)
download_quant_data
(
${
QUANT2_MobileNetV1_MODEL_DIR
}
"MobileNet_qat_perf.tar.gz"
)
download_quant_data
_without_verify
(
${
QUANT2_MobileNetV1_MODEL_DIR
}
"MobileNet_qat_perf.tar.gz"
)
endif
(
NOT LINUX
)
endif
(
NOT LINUX
)
download_quant_data
(
${
QUANT2_INT8_MobileNetV1_MODEL_DIR
}
"MobileNet_qat_perf_int8.tar.gz"
)
download_quant_data
_without_verify
(
${
QUANT2_INT8_MobileNetV1_MODEL_DIR
}
"MobileNet_qat_perf_int8.tar.gz"
)
inference_analysis_api_quant_test_run
(
test_analyzer_quant_performance_benchmark
${
QUANT_IMG_CLASS_TEST_APP
}
${
QUANT2_MobileNetV1_MODEL_DIR
}
/MobileNet_qat_perf/float
${
QUANT2_INT8_MobileNetV1_MODEL_DIR
}
/MobileNet_qat_perf_int8
${
IMAGENET_DATA_PATH
}
)
inference_analysis_api_quant_test_run
(
test_analyzer_quant_performance_benchmark
${
QUANT_IMG_CLASS_TEST_APP
}
${
QUANT2_MobileNetV1_MODEL_DIR
}
/MobileNet_qat_perf/float
${
QUANT2_INT8_MobileNetV1_MODEL_DIR
}
/MobileNet_qat_perf_int8
${
IMAGENET_DATA_PATH
}
)
### Other tests
### Other tests
...
@@ -465,13 +505,13 @@ if(WITH_MKLDNN)
...
@@ -465,13 +505,13 @@ if(WITH_MKLDNN)
inference_analysis_test_run
(
test_mkldnn_quantizer_config COMMAND
${
MKLDNN_QUANTIZER_CONFIG_TEST_APP
}
)
inference_analysis_test_run
(
test_mkldnn_quantizer_config COMMAND
${
MKLDNN_QUANTIZER_CONFIG_TEST_APP
}
)
# preprocess data2bin imagenet
# preprocess data2bin imagenet
download_int8_data
(
${
INT8_DATA_DIR
}
"imagenet_small.tar.gz"
)
download_int8_data
_without_verify
(
${
INT8_DATA_DIR
}
"imagenet_small.tar.gz"
)
set
(
IMAGENET_SMALL_DATA_DIR
"
${
INT8_DATA_DIR
}
/imagenet_small"
)
set
(
IMAGENET_SMALL_DATA_DIR
"
${
INT8_DATA_DIR
}
/imagenet_small"
)
set
(
IMAGENET_SMALL_OUTPUT_FILE
"imagenet_small.bin"
)
set
(
IMAGENET_SMALL_OUTPUT_FILE
"imagenet_small.bin"
)
preprocess_data2bin_test_run
(
preprocess_local_imagenet
"full_ILSVRC2012_val_preprocess.py"
${
IMAGENET_SMALL_DATA_DIR
}
${
IMAGENET_SMALL_OUTPUT_FILE
}
)
preprocess_data2bin_test_run
(
preprocess_local_imagenet
"full_ILSVRC2012_val_preprocess.py"
${
IMAGENET_SMALL_DATA_DIR
}
${
IMAGENET_SMALL_OUTPUT_FILE
}
)
# preprocess data2bin pascalvoc
# preprocess data2bin pascalvoc
download_int8_data
(
${
INT8_DATA_DIR
}
"pascalvoc_small.tar.gz"
)
download_int8_data
_without_verify
(
${
INT8_DATA_DIR
}
"pascalvoc_small.tar.gz"
)
set
(
PASCALVOC_SMALL_DATA_DIR
"
${
INT8_DATA_DIR
}
/pascalvoc_small"
)
set
(
PASCALVOC_SMALL_DATA_DIR
"
${
INT8_DATA_DIR
}
/pascalvoc_small"
)
set
(
PASCALVOC_SMALL_OUTPUT_FILE
"pascalvoc_small.bin"
)
set
(
PASCALVOC_SMALL_OUTPUT_FILE
"pascalvoc_small.bin"
)
preprocess_data2bin_test_run
(
preprocess_local_pascalvoc
"full_pascalvoc_test_preprocess.py"
${
PASCALVOC_SMALL_DATA_DIR
}
${
PASCALVOC_SMALL_OUTPUT_FILE
}
)
preprocess_data2bin_test_run
(
preprocess_local_pascalvoc
"full_pascalvoc_test_preprocess.py"
${
PASCALVOC_SMALL_DATA_DIR
}
${
PASCALVOC_SMALL_OUTPUT_FILE
}
)
...
@@ -480,26 +520,26 @@ endif()
...
@@ -480,26 +520,26 @@ endif()
# bert, max_len=20, embedding_dim=128
# bert, max_len=20, embedding_dim=128
set
(
BERT_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/bert_emb128"
)
set
(
BERT_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/bert_emb128"
)
download_model_and_data
(
${
BERT_INSTALL_DIR
}
"bert_emb128_model.tar.gz"
"bert_data_len20.txt.tar.gz"
)
download_model_and_data
_without_verify
(
${
BERT_INSTALL_DIR
}
"bert_emb128_model.tar.gz"
"bert_data_len20.txt.tar.gz"
)
inference_analysis_api_test
(
test_analyzer_bert
${
BERT_INSTALL_DIR
}
analyzer_bert_tester.cc
)
inference_analysis_api_test
(
test_analyzer_bert
${
BERT_INSTALL_DIR
}
analyzer_bert_tester.cc
)
# multiple models prediction
# multiple models prediction
set
(
MMP_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/multi_model_prediction"
)
set
(
MMP_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/multi_model_prediction"
)
download_data
(
${
MMP_INSTALL_DIR
}
PaddleInference/mobilenet_v2_models.tar.gz
)
download_data
_without_verify
(
${
MMP_INSTALL_DIR
}
PaddleInference/mobilenet_v2_models.tar.gz
)
inference_multiple_models_analysis_api_test
(
test_analyzer_multi_model_prediction
${
MMP_INSTALL_DIR
}
analyzer_mmp_tester.cc
)
inference_multiple_models_analysis_api_test
(
test_analyzer_multi_model_prediction
${
MMP_INSTALL_DIR
}
analyzer_mmp_tester.cc
)
if
(
WITH_GPU AND TENSORRT_FOUND
)
if
(
WITH_GPU AND TENSORRT_FOUND
)
set
(
TRT_MODEL_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/trt_models"
)
set
(
TRT_MODEL_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/trt_models"
)
if
(
NOT EXISTS
${
TRT_MODEL_INSTALL_DIR
}
/trt_inference_test_models.tar.gz
)
if
(
NOT EXISTS
${
TRT_MODEL_INSTALL_DIR
}
/trt_inference_test_models.tar.gz
)
inference_download_and_uncompress
(
${
TRT_MODEL_INSTALL_DIR
}
${
INFERENCE_URL
}
/tensorrt_test
"trt_inference_test_models.tar.gz"
)
inference_download_and_uncompress
(
${
TRT_MODEL_INSTALL_DIR
}
${
INFERENCE_URL
}
/tensorrt_test
"trt_inference_test_models.tar.gz"
3dcccdc38b549b6b1b4089723757bd98
)
endif
()
endif
()
set
(
TEST_SPLIT_CONVERTER_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/trt_split_op_converter_test"
)
set
(
TEST_SPLIT_CONVERTER_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/trt_split_op_converter_test"
)
if
(
NOT EXISTS
${
TEST_SPLIT_CONVERTER_MODEL
}
/split_converter.tgz
)
if
(
NOT EXISTS
${
TEST_SPLIT_CONVERTER_MODEL
}
/split_converter.tgz
)
inference_download_and_uncompress
(
${
TEST_SPLIT_CONVERTER_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"split_converter.tgz"
)
inference_download_and_uncompress
_without_verify
(
${
TEST_SPLIT_CONVERTER_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"split_converter.tgz"
)
endif
()
endif
()
set
(
TEST_INSTANCE_NORM_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/trt_instance_norm_test"
)
set
(
TEST_INSTANCE_NORM_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/trt_instance_norm_test"
)
if
(
NOT EXISTS
${
TEST_INSTANCE_NORM_MODEL
}
/instance_norm.tgz
)
if
(
NOT EXISTS
${
TEST_INSTANCE_NORM_MODEL
}
/instance_norm.tgz
)
inference_download_and_uncompress
(
${
TEST_INSTANCE_NORM_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"instance_norm.tgz"
)
inference_download_and_uncompress
_without_verify
(
${
TEST_INSTANCE_NORM_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"instance_norm.tgz"
)
endif
()
endif
()
inference_analysis_test
(
trt_mobilenet_test SRCS trt_mobilenet_test.cc
inference_analysis_test
(
trt_mobilenet_test SRCS trt_mobilenet_test.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
...
@@ -531,7 +571,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
...
@@ -531,7 +571,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
set
(
TRT_MODEL_QUANT_RESNET_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/small_quant_model"
)
set
(
TRT_MODEL_QUANT_RESNET_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/small_quant_model"
)
if
(
NOT EXISTS
${
INFERENCE_DEMO_INSTALL_DIR
}
/small_quant_model.tgz
)
if
(
NOT EXISTS
${
INFERENCE_DEMO_INSTALL_DIR
}
/small_quant_model.tgz
)
inference_download_and_uncompress
(
${
INFERENCE_DEMO_INSTALL_DIR
}
${
INFERENCE_URL
}
/tensorrt_test
"small_quant_model.tgz"
)
inference_download_and_uncompress
_without_verify
(
${
INFERENCE_DEMO_INSTALL_DIR
}
${
INFERENCE_URL
}
/tensorrt_test
"small_quant_model.tgz"
)
endif
()
endif
()
inference_analysis_test
(
trt_quant_int8_test SRCS trt_quant_int8_test.cc
inference_analysis_test
(
trt_quant_int8_test SRCS trt_quant_int8_test.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
...
@@ -539,7 +579,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
...
@@ -539,7 +579,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
set
(
TRT_MODEL_QUANT_YOLOV3_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/yolov3_r50_quant_aware"
)
set
(
TRT_MODEL_QUANT_YOLOV3_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/yolov3_r50_quant_aware"
)
if
(
NOT EXISTS
${
INFERENCE_DEMO_INSTALL_DIR
}
/yolov3_r50_quant_aware.tgz
)
if
(
NOT EXISTS
${
INFERENCE_DEMO_INSTALL_DIR
}
/yolov3_r50_quant_aware.tgz
)
inference_download_and_uncompress
(
${
INFERENCE_DEMO_INSTALL_DIR
}
${
INFERENCE_URL
}
/tensorrt_test
"yolov3_r50_quant_aware.tgz"
)
inference_download_and_uncompress
_without_verify
(
${
INFERENCE_DEMO_INSTALL_DIR
}
${
INFERENCE_URL
}
/tensorrt_test
"yolov3_r50_quant_aware.tgz"
)
endif
()
endif
()
inference_analysis_test
(
trt_quant_int8_yolov3_r50_test SRCS trt_quant_int8_yolov3_r50_test.cc
inference_analysis_test
(
trt_quant_int8_yolov3_r50_test SRCS trt_quant_int8_yolov3_r50_test.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
...
@@ -547,12 +587,12 @@ if(WITH_GPU AND TENSORRT_FOUND)
...
@@ -547,12 +587,12 @@ if(WITH_GPU AND TENSORRT_FOUND)
set
(
TEST_TRT_DYNAMIC_MODEL2
"
${
TRT_MODEL_INSTALL_DIR
}
/complex_model_dynamic"
)
set
(
TEST_TRT_DYNAMIC_MODEL2
"
${
TRT_MODEL_INSTALL_DIR
}
/complex_model_dynamic"
)
if
(
NOT EXISTS
${
TEST_TRT_DYNAMIC_MODEL2
}
/complex_model_dynamic2.tar.gz
)
if
(
NOT EXISTS
${
TEST_TRT_DYNAMIC_MODEL2
}
/complex_model_dynamic2.tar.gz
)
inference_download_and_uncompress
(
${
TEST_TRT_DYNAMIC_MODEL2
}
${
INFERENCE_URL
}
/tensorrt_test
"complex_model_dynamic2.tar.gz"
)
inference_download_and_uncompress
_without_verify
(
${
TEST_TRT_DYNAMIC_MODEL2
}
${
INFERENCE_URL
}
/tensorrt_test
"complex_model_dynamic2.tar.gz"
)
endif
()
endif
()
set
(
TEST_TRT_DYNAMIC_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/conv_bn_swish_split_gelu"
)
set
(
TEST_TRT_DYNAMIC_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/conv_bn_swish_split_gelu"
)
if
(
NOT EXISTS
${
TEST_TRT_DYNAMIC_MODEL
}
/conv_bn_swish_split_gelu.tar.gz
)
if
(
NOT EXISTS
${
TEST_TRT_DYNAMIC_MODEL
}
/conv_bn_swish_split_gelu.tar.gz
)
inference_download_and_uncompress
(
${
TEST_TRT_DYNAMIC_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"conv_bn_swish_split_gelu.tar.gz"
)
inference_download_and_uncompress
(
${
TEST_TRT_DYNAMIC_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"conv_bn_swish_split_gelu.tar.gz"
2a5e8791e47b221b4f782151d76da9c6
)
endif
()
endif
()
inference_analysis_test
(
trt_dynamic_shape_test SRCS trt_dynamic_shape_test.cc
inference_analysis_test
(
trt_dynamic_shape_test SRCS trt_dynamic_shape_test.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
...
@@ -560,7 +600,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
...
@@ -560,7 +600,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
set
(
TEST_TRT_ERNIE_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/ernie_test"
)
set
(
TEST_TRT_ERNIE_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/ernie_test"
)
if
(
NOT EXISTS
${
TEST_TRT_ERNIE_MODEL
}
/ernie_model_4.tar.gz
)
if
(
NOT EXISTS
${
TEST_TRT_ERNIE_MODEL
}
/ernie_model_4.tar.gz
)
inference_download_and_uncompress
(
${
TEST_TRT_ERNIE_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"ernie_model_4.tar.gz"
)
inference_download_and_uncompress
(
${
TEST_TRT_ERNIE_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"ernie_model_4.tar.gz"
5fa371efa75706becbaad79195d2ca68
)
endif
()
endif
()
inference_analysis_test
(
test_trt_dynamic_shape_ernie SRCS trt_dynamic_shape_ernie_test.cc
inference_analysis_test
(
test_trt_dynamic_shape_ernie SRCS trt_dynamic_shape_ernie_test.cc
...
@@ -569,7 +609,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
...
@@ -569,7 +609,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
set
(
TEST_TRT_TRANSFORMER_PRUNE_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/transformer_prune"
)
set
(
TEST_TRT_TRANSFORMER_PRUNE_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/transformer_prune"
)
if
(
NOT EXISTS
${
TEST_TRT_TRANSFORMER_PRUNE_MODEL
}
/transformer_prune.tar.gz
)
if
(
NOT EXISTS
${
TEST_TRT_TRANSFORMER_PRUNE_MODEL
}
/transformer_prune.tar.gz
)
inference_download_and_uncompress
(
${
TEST_TRT_TRANSFORMER_PRUNE_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"transformer_prune.tar.gz"
)
inference_download_and_uncompress
(
${
TEST_TRT_TRANSFORMER_PRUNE_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"transformer_prune.tar.gz"
77b56dc73ff0cf44ddb1ce9ca0b0f471
)
endif
()
endif
()
inference_analysis_test
(
test_trt_dynamic_shape_transformer_prune SRCS trt_dynamic_shape_transformer_prune_test.cc
inference_analysis_test
(
test_trt_dynamic_shape_transformer_prune SRCS trt_dynamic_shape_transformer_prune_test.cc
...
@@ -577,7 +617,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
...
@@ -577,7 +617,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
ARGS --infer_model=
${
TEST_TRT_TRANSFORMER_PRUNE_MODEL
}
/transformer_prune
)
ARGS --infer_model=
${
TEST_TRT_TRANSFORMER_PRUNE_MODEL
}
/transformer_prune
)
if
(
NOT EXISTS
${
TEST_TRT_ERNIE_MODEL
}
/ernie_model_4_unserialized.tgz
)
if
(
NOT EXISTS
${
TEST_TRT_ERNIE_MODEL
}
/ernie_model_4_unserialized.tgz
)
inference_download_and_uncompress
(
${
TEST_TRT_ERNIE_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"ernie_model_4_unserialized.tgz"
)
inference_download_and_uncompress
(
${
TEST_TRT_ERNIE_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"ernie_model_4_unserialized.tgz"
833d73fc6a7f7e1ee4a1fd6419209e55
)
endif
()
endif
()
inference_analysis_test
(
test_trt_dynamic_shape_ernie_ser_deser SRCS trt_dynamic_shape_ernie_serialize_deserialize_test.cc
inference_analysis_test
(
test_trt_dynamic_shape_ernie_ser_deser SRCS trt_dynamic_shape_ernie_serialize_deserialize_test.cc
...
@@ -585,7 +625,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
...
@@ -585,7 +625,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
ARGS --infer_model=
${
TEST_TRT_ERNIE_MODEL
}
/ernie_model_4_unserialized
)
ARGS --infer_model=
${
TEST_TRT_ERNIE_MODEL
}
/ernie_model_4_unserialized
)
if
(
NOT EXISTS
${
TEST_TRT_ERNIE_MODEL
}
/ernie_model_4_fp16_unserialized.tgz
)
if
(
NOT EXISTS
${
TEST_TRT_ERNIE_MODEL
}
/ernie_model_4_fp16_unserialized.tgz
)
inference_download_and_uncompress
(
${
TEST_TRT_ERNIE_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"ernie_model_4_fp16_unserialized.tgz"
)
inference_download_and_uncompress
(
${
TEST_TRT_ERNIE_MODEL
}
${
INFERENCE_URL
}
/tensorrt_test
"ernie_model_4_fp16_unserialized.tgz"
c5ff2d0cad79953ffbf2b8b9e2fae6e4
)
endif
()
endif
()
inference_analysis_test
(
test_trt_dynamic_shape_ernie_fp16_ser_deser SRCS trt_dynamic_shape_ernie_fp16_serialize_deserialize_test.cc
inference_analysis_test
(
test_trt_dynamic_shape_ernie_fp16_ser_deser SRCS trt_dynamic_shape_ernie_fp16_serialize_deserialize_test.cc
...
@@ -595,7 +635,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
...
@@ -595,7 +635,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
endif
()
endif
()
set
(
LITE_MODEL_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/lite"
)
set
(
LITE_MODEL_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/lite"
)
download_data
(
${
LITE_MODEL_INSTALL_DIR
}
"mul_model_fp32.tgz"
)
download_data
_without_verify
(
${
LITE_MODEL_INSTALL_DIR
}
"mul_model_fp32.tgz"
)
inference_analysis_test
(
lite_mul_model_test SRCS lite_mul_model_test.cc
inference_analysis_test
(
lite_mul_model_test SRCS lite_mul_model_test.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
...
...
paddle/fluid/inference/tests/test.cmake
浏览文件 @
b6267919
...
@@ -23,7 +23,30 @@ function(inference_download INSTALL_DIR URL FILENAME)
...
@@ -23,7 +23,30 @@ function(inference_download INSTALL_DIR URL FILENAME)
)
)
endfunction
()
endfunction
()
function
(
inference_download_and_uncompress INSTALL_DIR URL FILENAME
)
function
(
inference_download_and_uncompress INSTALL_DIR URL FILENAME CHECK_SUM
)
message
(
STATUS
"Download inference test stuff from
${
URL
}
/
${
FILENAME
}
"
)
string
(
REGEX REPLACE
"[-%./
\\
]"
"_"
FILENAME_EX
${
FILENAME
}
)
string
(
REGEX MATCH
"[^/
\\
]+$"
DOWNLOAD_NAME
${
FILENAME
}
)
set
(
EXTERNAL_PROJECT_NAME
"extern_download_
${
FILENAME_EX
}
"
)
set
(
UNPACK_DIR
"
${
INSTALL_DIR
}
/src/
${
EXTERNAL_PROJECT_NAME
}
"
)
ExternalProject_Add
(
${
EXTERNAL_PROJECT_NAME
}
${
EXTERNAL_PROJECT_LOG_ARGS
}
PREFIX
${
INSTALL_DIR
}
URL
${
URL
}
/
${
FILENAME
}
URL_HASH MD5=
${
CHECK_SUM
}
DOWNLOAD_DIR
${
INSTALL_DIR
}
DOWNLOAD_NO_EXTRACT 1
DOWNLOAD_NO_PROGRESS 1
CONFIGURE_COMMAND
""
BUILD_COMMAND
${
CMAKE_COMMAND
}
-E chdir
${
INSTALL_DIR
}
${
CMAKE_COMMAND
}
-E tar xzf
${
DOWNLOAD_NAME
}
UPDATE_COMMAND
""
INSTALL_COMMAND
""
)
endfunction
()
function
(
inference_download_and_uncompress_without_verify INSTALL_DIR URL FILENAME
)
message
(
STATUS
"Download inference test stuff from
${
URL
}
/
${
FILENAME
}
"
)
message
(
STATUS
"Download inference test stuff from
${
URL
}
/
${
FILENAME
}
"
)
string
(
REGEX REPLACE
"[-%./
\\
]"
"_"
FILENAME_EX
${
FILENAME
}
)
string
(
REGEX REPLACE
"[-%./
\\
]"
"_"
FILENAME_EX
${
FILENAME
}
)
string
(
REGEX MATCH
"[^/
\\
]+$"
DOWNLOAD_NAME
${
FILENAME
}
)
string
(
REGEX MATCH
"[^/
\\
]+$"
DOWNLOAD_NAME
${
FILENAME
}
)
...
@@ -47,13 +70,13 @@ endfunction()
...
@@ -47,13 +70,13 @@ endfunction()
set
(
WORD2VEC_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/word2vec"
)
set
(
WORD2VEC_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/word2vec"
)
if
(
NOT EXISTS
${
WORD2VEC_INSTALL_DIR
}
/word2vec.inference.model.tar.gz
)
if
(
NOT EXISTS
${
WORD2VEC_INSTALL_DIR
}
/word2vec.inference.model.tar.gz
)
inference_download_and_uncompress
(
${
WORD2VEC_INSTALL_DIR
}
${
INFERENCE_URL
}
"word2vec.inference.model.tar.gz"
)
inference_download_and_uncompress
_without_verify
(
${
WORD2VEC_INSTALL_DIR
}
${
INFERENCE_URL
}
"word2vec.inference.model.tar.gz"
)
endif
()
endif
()
set
(
WORD2VEC_MODEL_DIR
"
${
WORD2VEC_INSTALL_DIR
}
/word2vec.inference.model"
)
set
(
WORD2VEC_MODEL_DIR
"
${
WORD2VEC_INSTALL_DIR
}
/word2vec.inference.model"
)
set
(
IMG_CLS_RESNET_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/image_classification_resnet"
)
set
(
IMG_CLS_RESNET_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/image_classification_resnet"
)
if
(
NOT EXISTS
${
IMG_CLS_RESNET_INSTALL_DIR
}
/image_classification_resnet.inference.model.tgz
)
if
(
NOT EXISTS
${
IMG_CLS_RESNET_INSTALL_DIR
}
/image_classification_resnet.inference.model.tgz
)
inference_download_and_uncompress
(
${
IMG_CLS_RESNET_INSTALL_DIR
}
${
INFERENCE_URL
}
"image_classification_resnet.inference.model.tgz"
)
inference_download_and_uncompress
_without_verify
(
${
IMG_CLS_RESNET_INSTALL_DIR
}
${
INFERENCE_URL
}
"image_classification_resnet.inference.model.tgz"
)
endif
()
endif
()
set
(
IMG_CLS_RESNET_MODEL_DIR
"
${
IMG_CLS_RESNET_INSTALL_DIR
}
/image_classification_resnet.inference.model"
)
set
(
IMG_CLS_RESNET_MODEL_DIR
"
${
IMG_CLS_RESNET_INSTALL_DIR
}
/image_classification_resnet.inference.model"
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录