Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
Paddle
提交
b6267919
P
Paddle
项目概览
PaddlePaddle
/
Paddle
接近 2 年 前同步成功
通知
2323
Star
20933
Fork
5424
代码
文件
提交
分支
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看板
未验证
提交
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)
set
(
INFERENCE_EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
analysis
${
analysis_deps
}
)
endif
()
function
(
download_data install_dir data_file
)
function
(
download_data install_dir data_file
check_sum
)
string
(
REGEX MATCH
"[^/
\\
]+$"
file_name
${
data_file
}
)
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
()
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
}
)
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
()
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
}
)
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
()
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
}
)
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
()
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
}
)
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
()
endfunction
()
function
(
download_model_and_data install_dir model_name data_name
)
download_data
(
${
install_dir
}
${
model_name
}
)
download_data
(
${
install_dir
}
${
data_name
}
)
function
(
download_quant_data install_dir data_file check_sum
)
if
(
NOT EXISTS
${
install_dir
}
/
${
data_file
}
)
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
()
function
(
download_result install_dir result_name
)
download_data
(
${
install_dir
}
${
result_name
}
)
function
(
download_result
_without_verify
install_dir result_name
)
download_data
_without_verify
(
${
install_dir
}
${
result_name
}
)
endfunction
()
function
(
inference_analysis_api_test target install_dir filename
)
...
...
@@ -165,12 +205,12 @@ endfunction()
if
(
NOT APPLE AND WITH_MKLML
)
# 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
)
# seq_pool1
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
${
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
)
...
...
@@ -193,7 +233,7 @@ endif()
# 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
)
# TODO(luotao, Superjom) Disable DAM test, temporarily fix
...
...
@@ -201,12 +241,12 @@ inference_analysis_api_test(test_analyzer_rnn2 ${RNN2_INSTALL_DIR} analyzer_rnn2
# After inference framework refactor, will reopen it.
# normal 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)
# 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
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
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
# 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
)
# 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
)
# 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
)
#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_result
(
${
ERNIE_INSTALL_DIR
}
"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"
73beea65abda2edb61c1662cd3180c62
)
inference_analysis_api_test
(
test_analyzer_ernie
${
ERNIE_INSTALL_DIR
}
analyzer_ernie_tester.cc
)
#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_result
(
${
ERNIE_INSTALL_DIR
}
"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"
1facda98eef1085dc9d435ebf3f23a73
)
inference_analysis_test
(
test_analyzer_ernie_large SRCS analyzer_ernie_tester.cc
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
)
...
...
@@ -251,17 +291,17 @@ endif()
# 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
)
# 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
)
# transformer, the dataset only works on batch_size=8 now
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
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
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
# ocr
set
(
OCR_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/ocr"
)
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
()
inference_analysis_api_test
(
test_analyzer_ocr
${
OCR_INSTALL_DIR
}
analyzer_vis_tester.cc
)
# 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
# EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
# 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")
# mobilenet with transpose op
set
(
MOBILENET_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/mobilenet"
)
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
()
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
# 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
}
${
GOOGLENET_MODEL_DIR
}
false
)
# 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
}
${
RESNET50_MODEL_DIR
}
true
)
if
(
WIN32
)
...
...
@@ -323,7 +363,7 @@ endif()
# mobilenet with depthwise_conv op
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
}
${
MOBILENET_MODEL_DIR
}
false
)
...
...
@@ -340,7 +380,7 @@ if(WITH_MKLDNN)
set
(
IMAGENET_DATA_ARCHIVE
"imagenet_val_100_tail.tar.gz"
)
set
(
IMAGENET_DATA_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/imagenet"
)
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
set
(
INT8_IMG_CLASS_TEST_APP
"test_analyzer_int8_image_classification"
)
...
...
@@ -349,40 +389,40 @@ if(WITH_MKLDNN)
# resnet50 int8
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
}
)
# mobilenetv1 int8
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
}
)
# mobilenetv2 int8
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
}
)
# resnet101 int8
# TODO(grygielski) Enable after MKL-DNN 1.0 merge
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})
# vgg16 int8
# TODO(grygielski) Enable after MKL-DNN 1.0 merge
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})
# vgg19 int8
# TODO(grygielski) Enable after MKL-DNN 1.0 merge
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})
# googlenet int8
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
)
### BFLOAT16 tests
...
...
@@ -410,7 +450,7 @@ if(WITH_MKLDNN)
set
(
INT8_OBJ_DETECT_TEST_APP_SRC
"analyzer_int8_object_detection_tester.cc"
)
# 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
...
...
@@ -418,13 +458,13 @@ if(WITH_MKLDNN)
# mobilenet-ssd int8
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
}
)
### Lexcial analysis GRU model
set
(
GRU_PATH
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/gru"
)
download_GRU_data
(
"
${
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_data.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_MODEL_PATH
"
${
GRU_PATH
}
/GRU_eval_model_v2"
)
set
(
LEXICAL_TEST_APP
"test_analyzer_lexical_analysis"
)
...
...
@@ -451,9 +491,9 @@ if(WITH_MKLDNN)
set
(
QUANT2_MobileNetV1_MODEL_DIR
"
${
QUANT_DATA_DIR
}
/MobileNetV1_quant2"
)
set
(
QUANT2_INT8_MobileNetV1_MODEL_DIR
"
${
QUANT_DATA_DIR
}
/MobileNetV1_quant2_int8"
)
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
)
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
}
)
### Other tests
...
...
@@ -465,13 +505,13 @@ if(WITH_MKLDNN)
inference_analysis_test_run
(
test_mkldnn_quantizer_config COMMAND
${
MKLDNN_QUANTIZER_CONFIG_TEST_APP
}
)
# 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_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 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_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
}
)
...
...
@@ -480,26 +520,26 @@ endif()
# bert, max_len=20, embedding_dim=128
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
)
# multiple models 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
)
if
(
WITH_GPU AND TENSORRT_FOUND
)
set
(
TRT_MODEL_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/trt_models"
)
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
()
set
(
TEST_SPLIT_CONVERTER_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/trt_split_op_converter_test"
)
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
()
set
(
TEST_INSTANCE_NORM_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/trt_instance_norm_test"
)
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
()
inference_analysis_test
(
trt_mobilenet_test SRCS trt_mobilenet_test.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
...
...
@@ -531,7 +571,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
set
(
TRT_MODEL_QUANT_RESNET_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/small_quant_model"
)
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
()
inference_analysis_test
(
trt_quant_int8_test SRCS trt_quant_int8_test.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
...
...
@@ -539,7 +579,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
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
)
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
()
inference_analysis_test
(
trt_quant_int8_yolov3_r50_test SRCS trt_quant_int8_yolov3_r50_test.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
...
...
@@ -547,12 +587,12 @@ if(WITH_GPU AND TENSORRT_FOUND)
set
(
TEST_TRT_DYNAMIC_MODEL2
"
${
TRT_MODEL_INSTALL_DIR
}
/complex_model_dynamic"
)
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
()
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
)
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
()
inference_analysis_test
(
trt_dynamic_shape_test SRCS trt_dynamic_shape_test.cc
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
...
...
@@ -560,7 +600,7 @@ if(WITH_GPU AND TENSORRT_FOUND)
set
(
TEST_TRT_ERNIE_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/ernie_test"
)
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
()
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)
set
(
TEST_TRT_TRANSFORMER_PRUNE_MODEL
"
${
TRT_MODEL_INSTALL_DIR
}
/transformer_prune"
)
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
()
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)
ARGS --infer_model=
${
TEST_TRT_TRANSFORMER_PRUNE_MODEL
}
/transformer_prune
)
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
()
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)
ARGS --infer_model=
${
TEST_TRT_ERNIE_MODEL
}
/ernie_model_4_unserialized
)
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
()
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)
endif
()
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
EXTRA_DEPS
${
INFERENCE_EXTRA_DEPS
}
...
...
paddle/fluid/inference/tests/test.cmake
浏览文件 @
b6267919
...
...
@@ -23,7 +23,30 @@ function(inference_download INSTALL_DIR URL FILENAME)
)
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
}
"
)
string
(
REGEX REPLACE
"[-%./
\\
]"
"_"
FILENAME_EX
${
FILENAME
}
)
string
(
REGEX MATCH
"[^/
\\
]+$"
DOWNLOAD_NAME
${
FILENAME
}
)
...
...
@@ -47,13 +70,13 @@ endfunction()
set
(
WORD2VEC_INSTALL_DIR
"
${
INFERENCE_DEMO_INSTALL_DIR
}
/word2vec"
)
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
()
set
(
WORD2VEC_MODEL_DIR
"
${
WORD2VEC_INSTALL_DIR
}
/word2vec.inference.model"
)
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
)
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
()
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.
先完成此消息的编辑！
取消
想要评论请
注册
或
登录
