Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
Paddle
提交
b6267919
P
Paddle
项目概览
BaiXuePrincess
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
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.
先完成此消息的编辑！
取消
想要评论请
注册
或
登录
