@ -1,77 +1,87 @@
set ( INFERENCE_URL "http://paddle-inference-dist.bj.bcebos.com" )
set ( INFERENCE_DEMO_INSTALL_DIR "${THIRD_PARTY_PATH}/inference_demo" )
set ( INFERENCE_URL "http://paddle-inference-dist.cdn.bcebos.com" )
set ( INFERENCE_DEMO_INSTALL_DIR "${THIRD_PARTY_PATH}/inference_demo" CACHE STRING
" A p a t h s e t t i n g i n f e r e n c e d e m o d o w n l o a d d i r e c t o r i e s . " )
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}")
function ( inference_download install_dir url filename )
message ( STATUS "Download inference test stuff from ${ url }/${filename}")
execute_process ( COMMAND bash -c "mkdir -p ${install_dir}" )
execute_process ( COMMAND bash -c "cd ${install_dir} && wget -q ${INFERENCE_URL}/${filename}" )
execute_process ( COMMAND bash -c "cd ${install_dir} && tar xzf ${filename}" )
execute_process ( COMMAND bash -c "cd ${install_dir} && wget -q ${url}/${filename}" )
message ( STATUS "finish downloading ${filename}" )
endfunction ( inference_download_and_uncompress )
endfunction ( )
function ( inference_download_and_uncompress install_dir url filename )
inference_download ( ${ install_dir } ${ url } ${ filename } )
execute_process ( COMMAND bash -c "cd ${install_dir} && tar xzf ${filename}" )
endfunction ( )
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_name } )
inference_download_and_uncompress ( ${ install_dir } ${ data_name } )
if ( NOT EXISTS ${ install_dir } )
inference_download_and_uncompress ( ${ install_dir } ${ INFERENCE_URL} ${ model_name} )
inference_download_and_uncompress ( ${ install_dir } ${ INFERENCE_URL} ${ data_name} )
endif ( )
endfunction ( )
function ( inference_analysis_api_test target install_dir filename )
inference_analysis_test ( ${ target } SRCS ${ filename }
E X T R A _ D E P S $ { I N F E R E N C E _ E X T R A _ D E P S }
A R G S - - i n f e r _ m o d e l = $ { i n s t a l l _ d i r } / m o d e l - - i n f e r _ d a t a = $ { i n s t a l l _ d i r } / d a t a . t x t )
endfunction ( )
# R N N 1
# T O D O : f i x t h i s t e s t o n M A C O S
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
E X T R A _ D E P S $ { I N F E R E N C E _ E X T R A _ D E P S }
A R G S - - i n f e r _ m o d e l = $ { R N N 1 _ I N S T A L L _ D I R } / m o d e l
- - i n f e r _ d a t a = $ { R N N 1 _ I N S T A L L _ D I R } / d a t a . t x t )
endif ( NOT APPLE )
inference_analysis_api_test ( test_analyzer_rnn1 ${ RNN1_INSTALL_DIR } analyzer_rnn1_tester.cc )
else ( )
# T O D O : f i x t h i s t e s t o n M A C O S , t h e r e a s o n i s t h a t
# f u s i o n _ s e q e x p a n d _ c o n c a t _ f c _ o p i s n o t s u p p o r t e d o n M A C O S
message ( WARNING "These tests has been disabled in OSX before being fixed: \n test_analyzer_rnn1" )
endif ( )
# R N N 2
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
E X T R A _ D E P S $ { I N F E R E N C E _ E X T R A _ D E P S }
A R G S - - i n f e r _ m o d e l = $ { R N N 2 _ I N S T A L L _ D I R } / m o d e l
- - i n f e r _ d a t a = $ { R N N 2 _ I N S T A L L _ D I R } / d a t a . t x t )
inference_analysis_api_test ( test_analyzer_rnn2 ${ RNN2_INSTALL_DIR } analyzer_rnn2_tester.cc )
# c h i n e s e _ n e r
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
E X T R A _ D E P S $ { I N F E R E N C E _ E X T R A _ D E P S }
A R G S - - i n f e r _ m o d e l = $ { C H I N E S E _ N E R _ I N S T A L L _ D I R } / m o d e l
- - i n f e r _ d a t a = $ { C H I N E S E _ N E R _ I N S T A L L _ D I R } / d a t a . t x t )
inference_analysis_api_test ( test_analyzer_ner ${ CHINESE_NER_INSTALL_DIR } analyzer_ner_tester.cc )
# l a c
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
E X T R A _ D E P S $ { I N F E R E N C E _ E X T R A _ D E P S }
A R G S - - i n f e r _ m o d e l = $ { L A C _ I N S T A L L _ D I R } / m o d e l
- - i n f e r _ d a t a = $ { L A C _ I N S T A L L _ D I R } / d a t a . t x t )
inference_analysis_api_test ( test_analyzer_lac ${ LAC_INSTALL_DIR } analyzer_lac_tester.cc )
# t e x t _ c l a s s i f i c a t i o n
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
E X T R A _ D E P S $ { I N F E R E N C E _ E X T R A _ D E P S }
A R G S - - i n f e r _ m o d e l = $ { T E X T _ C L A S S I F I C A T I O N _ I N S T A L L _ D I R } / m o d e l
- - i n f e r _ d a t a = $ { T E X T _ C L A S S I F I C A T I O N _ I N S T A L L _ D I R } / d a t a . t x t )
inference_analysis_api_test ( test_analyzer_text_classification ${ TEXT_CLASSIFICATION_INSTALL_DIR } analyzer_text_classification_tester.cc )
# o c r
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}" )
set ( OCR_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR}/ocr" )
if ( NOT EXISTS ${ OCR_INSTALL_DIR } )
inference_download_and_uncompress ( ${ OCR_INSTALL_DIR } "http://paddlemodels.cdn.bcebos.com/" "inference-vis-demos%2Focr.tar.gz" )
endif ( )
inference_analysis_api_test ( test_analyzer_ocr ${ OCR_INSTALL_DIR } analyzer_vis_tester.cc )
# a n a k i n
if ( WITH_ANAKIN AND WITH_MKL ) # o n l y n e e d e d i n C I
# a n a k i n r n n 1
set ( ANAKIN_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR}/anakin" )
set ( ANAKIN_RNN1_INSTALL_DIR "${ANAKIN_INSTALL_DIR}/rnn1" )
inference_download ( ${ ANAKIN_RNN1_INSTALL_DIR } ${ INFERENCE_URL } "anakin_test%2Fditu_rnn.anakin2.model.bin" )
inference_download ( ${ ANAKIN_RNN1_INSTALL_DIR } ${ INFERENCE_URL } "anakin_test%2Fditu_rnn_data.txt" )
cc_test ( test_anakin_rnn1 SRCS anakin_rnn1_tester.cc
A R G S - - m o d e l = $ { A N A K I N _ R N N 1 _ I N S T A L L _ D I R } / a n a k i n _ t e s t % 2 F d i t u _ r n n . a n a k i n 2 . m o d e l . b i n
- - d a t a p a t h = $ { A N A K I N _ R N N 1 _ I N S T A L L _ D I R } / a n a k i n _ t e s t % 2 F d i t u _ r n n _ d a t a . t x t
D E P S i n f e r e n c e _ a n a k i n _ a p i _ s h a r e d S E R I A L )
# a n a k i n m o b i l e n e t
if ( WITH_GPU )
set ( ANAKIN_MOBILENET_INSTALL_DIR "${ANAKIN_INSTALL_DIR}/mobilenet" )
inference_download ( ${ ANAKIN_MOBILENET_INSTALL_DIR } ${ INFERENCE_URL } "mobilenet_v2.anakin.bin" )
cc_test ( test_anakin_mobilenet SRCS anakin_mobilenet_tester.cc
A R G S - - m o d e l = $ { A N A K I N _ M O B I L E N E T _ I N S T A L L _ D I R } / m o b i l e n e t _ v 2 . a n a k i n . b i n
D E P S i n f e r e n c e _ a n a k i n _ a p i _ s h a r e d d y n l o a d _ c u d a S E R I A L )
endif ( )
endif ( )
inference_analysis_test ( test_analyzer_ocr SRCS analyzer_vis_tester.cc
E X T R A _ D E P S $ { I N F E R E N C E _ E X T R A _ D E P S }
A R G S - - i n f e r _ m o d e l = $ { O C R _ I N S T A L L _ D I R } / m o d e l
- - i n f e r _ d a t a = $ { O C R _ I N S T A L L _ D I R } / d a t a . t x t )
Tao Luo
7 years ago
committed by
Yan Chunwei