nhzlx
6 years ago
commit
cc4a7661c6
@ -1,57 +1,77 @@
|
||||
function (inference_download_and_uncompress install_dir url)
|
||||
get_filename_component(filename ${url} NAME)
|
||||
message(STATUS "Download inference test stuff ${filename} from ${url}")
|
||||
set(INFERENCE_URL "http://paddle-inference-dist.bj.bcebos.com")
|
||||
set(INFERENCE_DEMO_INSTALL_DIR "${THIRD_PARTY_PATH}/inference_demo")
|
||||
set(INFERENCE_EXTRA_DEPS paddle_inference_api paddle_fluid_api ir_pass_manager analysis_predictor)
|
||||
function (inference_download_and_uncompress install_dir filename)
|
||||
message(STATUS "Download inference test stuff from ${INFERENCE_URL}/${filename}")
|
||||
execute_process(COMMAND bash -c "mkdir -p ${install_dir}")
|
||||
execute_process(COMMAND bash -c "cd ${install_dir} && wget -q ${url}")
|
||||
execute_process(COMMAND bash -c "cd ${install_dir} && wget -q ${INFERENCE_URL}/${filename}")
|
||||
execute_process(COMMAND bash -c "cd ${install_dir} && tar xzf ${filename}")
|
||||
message(STATUS "finish downloading ${filename}")
|
||||
endfunction(inference_download_and_uncompress)
|
||||
|
||||
function(download_model_and_data install_dir model_url data_url)
|
||||
function(download_model_and_data install_dir model_name data_name)
|
||||
if (NOT EXISTS ${install_dir} AND WITH_INFERENCE)
|
||||
inference_download_and_uncompress(${install_dir} ${model_url})
|
||||
inference_download_and_uncompress(${install_dir} ${data_url})
|
||||
inference_download_and_uncompress(${install_dir} ${model_name})
|
||||
inference_download_and_uncompress(${install_dir} ${data_name})
|
||||
endif()
|
||||
endfunction()
|
||||
|
||||
# RNN1
|
||||
set(RNN1_MODEL_URL "http://paddle-inference-dist.bj.bcebos.com/rnn1%2Fmodel.tar.gz")
|
||||
set(RNN1_DATA_URL "http://paddle-inference-dist.bj.bcebos.com/rnn1%2Fdata.txt.tar.gz")
|
||||
set(RNN1_INSTALL_DIR "${THIRD_PARTY_PATH}/inference_demo/rnn1")
|
||||
download_model_and_data(${RNN1_INSTALL_DIR} ${RNN1_MODEL_URL} ${RNN1_DATA_URL})
|
||||
inference_analysis_test(test_analyzer_rnn1 SRCS analyzer_rnn1_tester.cc
|
||||
EXTRA_DEPS paddle_inference_api paddle_fluid_api ir_pass_manager analysis_predictor
|
||||
ARGS --infer_model=${RNN1_INSTALL_DIR}/model
|
||||
--infer_data=${RNN1_INSTALL_DIR}/data.txt)
|
||||
# TODO: fix this test on MACOS
|
||||
message(WARNING "These tests has been disabled in OSX before being fixed: \n test_analyzer_rnn1")
|
||||
if(NOT APPLE)
|
||||
set(RNN1_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR}/rnn1")
|
||||
download_model_and_data(${RNN1_INSTALL_DIR} "rnn1%2Fmodel.tar.gz" "rnn1%2Fdata.txt.tar.gz")
|
||||
inference_analysis_test(test_analyzer_rnn1 SRCS analyzer_rnn1_tester.cc
|
||||
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
|
||||
ARGS --infer_model=${RNN1_INSTALL_DIR}/model
|
||||
--infer_data=${RNN1_INSTALL_DIR}/data.txt)
|
||||
endif(NOT APPLE)
|
||||
|
||||
# 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")
|
||||
inference_analysis_test(test_analyzer_rnn2 SRCS analyzer_rnn2_tester.cc
|
||||
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
|
||||
ARGS --infer_model=${RNN2_INSTALL_DIR}/model
|
||||
--infer_data=${RNN2_INSTALL_DIR}/data.txt)
|
||||
|
||||
# chinese_ner
|
||||
set(CHINESE_NER_MODEL_URL "http://paddle-inference-dist.bj.bcebos.com/chinese_ner_model.tar.gz")
|
||||
set(CHINESE_NER_DATA_URL "http://paddle-inference-dist.bj.bcebos.com/chinese_ner-data.txt.tar.gz")
|
||||
set(CHINESE_NER_INSTALL_DIR "${THIRD_PARTY_PATH}/inference_demo/chinese_ner")
|
||||
download_model_and_data(${CHINESE_NER_INSTALL_DIR} ${CHINESE_NER_MODEL_URL} ${CHINESE_NER_DATA_URL})
|
||||
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")
|
||||
inference_analysis_test(test_analyzer_ner SRCS analyzer_ner_tester.cc
|
||||
EXTRA_DEPS paddle_inference_api paddle_fluid_api analysis_predictor
|
||||
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
|
||||
ARGS --infer_model=${CHINESE_NER_INSTALL_DIR}/model
|
||||
--infer_data=${CHINESE_NER_INSTALL_DIR}/data.txt)
|
||||
|
||||
# lac
|
||||
set(LAC_MODEL_URL "http://paddle-inference-dist.bj.bcebos.com/lac_model.tar.gz")
|
||||
set(LAC_DATA_URL "http://paddle-inference-dist.bj.bcebos.com/lac_data.txt.tar.gz")
|
||||
set(LAC_INSTALL_DIR "${THIRD_PARTY_PATH}/inference_demo/lac")
|
||||
download_model_and_data(${LAC_INSTALL_DIR} ${LAC_MODEL_URL} ${LAC_DATA_URL})
|
||||
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")
|
||||
inference_analysis_test(test_analyzer_lac SRCS analyzer_lac_tester.cc
|
||||
EXTRA_DEPS paddle_inference_api paddle_fluid_api ir_pass_manager analysis_predictor
|
||||
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
|
||||
ARGS --infer_model=${LAC_INSTALL_DIR}/model
|
||||
--infer_data=${LAC_INSTALL_DIR}/data.txt)
|
||||
|
||||
# text_classification
|
||||
set(TEXT_CLASSIFICATION_MODEL_URL "http://paddle-inference-dist.bj.bcebos.com/text-classification-Senta.tar.gz")
|
||||
set(TEXT_CLASSIFICATION_DATA_URL "http://paddle-inference-dist.bj.bcebos.com/text_classification_data.txt.tar.gz")
|
||||
set(TEXT_CLASSIFICATION_INSTALL_DIR "${THIRD_PARTY_PATH}/inference_demo/text_classification")
|
||||
download_model_and_data(${TEXT_CLASSIFICATION_INSTALL_DIR} ${TEXT_CLASSIFICATION_MODEL_URL} ${TEXT_CLASSIFICATION_DATA_URL})
|
||||
inference_analysis_test(test_text_classification SRCS analyzer_text_classification_tester.cc
|
||||
EXTRA_DEPS paddle_inference_api paddle_fluid_api analysis_predictor
|
||||
ARGS --infer_model=${TEXT_CLASSIFICATION_INSTALL_DIR}/text-classification-Senta
|
||||
--infer_data=${TEXT_CLASSIFICATION_INSTALL_DIR}/data.txt
|
||||
--topn=1 # Just run top 1 batch.
|
||||
)
|
||||
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")
|
||||
inference_analysis_test(test_analyzer_text_classification SRCS analyzer_text_classification_tester.cc
|
||||
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
|
||||
ARGS --infer_model=${TEXT_CLASSIFICATION_INSTALL_DIR}/model
|
||||
--infer_data=${TEXT_CLASSIFICATION_INSTALL_DIR}/data.txt)
|
||||
|
||||
# ocr
|
||||
set(OCR_MODEL_URL "http://paddlemodels.cdn.bcebos.com/inference-vis-demos%2Focr.tar.gz")
|
||||
set(OCR_INSTALL_DIR "${THIRD_PARTY_PATH}/inference_demo/ocr")
|
||||
if (NOT EXISTS ${OCR_INSTALL_DIR} AND WITH_INFERENCE)
|
||||
get_filename_component(filename ${OCR_MODEL_URL} NAME)
|
||||
message(STATUS "Download inference test stuff ${filename} from ${OCR_MODEL_URL}")
|
||||
execute_process(COMMAND bash -c "mkdir -p ${OCR_INSTALL_DIR}")
|
||||
execute_process(COMMAND bash -c "cd ${OCR_INSTALL_DIR} && wget -q ${OCR_MODEL_URL}")
|
||||
execute_process(COMMAND bash -c "cd ${OCR_INSTALL_DIR} && tar xzf ${filename}")
|
||||
message(STATUS "finish downloading ${filename}")
|
||||
endif()
|
||||
inference_analysis_test(test_analyzer_ocr SRCS analyzer_vis_tester.cc
|
||||
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
|
||||
ARGS --infer_model=${OCR_INSTALL_DIR}/model
|
||||
--infer_data=${OCR_INSTALL_DIR}/data.txt)
|
||||
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue