Skip to content

Commit 7ebff35

Browse files
committed
[tools] Add test model downloading scripts
Add test onnx models downloading scripts for future smoking test ONE-DCO-1.0-Signed-off-by: Youxin Chen <[email protected]>
1 parent 90f9c09 commit 7ebff35

File tree

2 files changed

+41
-0
lines changed

2 files changed

+41
-0
lines changed

tools/onnx_subgraph/CMakeLists.txt

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,3 +13,29 @@ find_package(Python3 COMPONENTS Interpreter Development REQUIRED)
1313

1414
include_directories(${CMAKE_CURRENT_BINARY_DIR})
1515
include_directories(${Python3_INCLUDE_DIRS})
16+
17+
set(ONNX_SUGRAPH_FILES
18+
test_model_download.sh
19+
)
20+
21+
foreach(ONNX_SUGRAPH IN ITEMS ${ONNX_SUGRAPH_FILES})
22+
set(ONNX_SUGRAPH_FILE ${ONNX_SUGRAPH})
23+
set(ONNX_SUGRAPH_SRC "${CMAKE_CURRENT_SOURCE_DIR}/${ONNX_SUGRAPH_FILE}")
24+
set(ONNX_SUGRAPH_BIN "${CMAKE_CURRENT_BINARY_DIR}/scripts/${ONNX_SUGRAPH_FILE}")
25+
set(ONNX_SUGRAPH_TARGET "${ONNX_SUGRAPH}_target")
26+
27+
add_custom_command(OUTPUT ${ONNX_SUGRAPH_BIN}
28+
COMMAND ${CMAKE_COMMAND} -E copy "${ONNX_SUGRAPH_SRC}" "${ONNX_SUGRAPH_BIN}"
29+
DEPENDS ${ONNX_SUGRAPH_SRC}
30+
COMMENT "Generate ${ONNX_SUGRAPH_BIN}"
31+
)
32+
33+
add_custom_target(${ONNX_SUGRAPH_TARGET} ALL DEPENDS ${ONNX_SUGRAPH_BIN})
34+
35+
install(FILES ${ONNX_SUGRAPH_BIN}
36+
PERMISSIONS OWNER_WRITE OWNER_READ OWNER_EXECUTE
37+
GROUP_READ GROUP_EXECUTE
38+
WORLD_READ WORLD_EXECUTE
39+
DESTINATION bin)
40+
41+
endforeach(ONNX_SUGRAPH)
Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
pip install onnx onnxsim
2+
3+
if [ ! -d "./models/" ];then
4+
mkdir ./models/
5+
else
6+
echo "./models path existing"
7+
fi
8+
9+
cd ./models
10+
wget https://media.githubusercontent.com/media/onnx/models/refs/heads/main/Computer_Vision/resnext26ts_Opset16_timm/resnext26ts_Opset16.onnx --no-check-certificate
11+
#wget https://media.githubusercontent.com/media/onnx/models/refs/heads/main/Natural_Language_Processing/xmod_Opset16_transformers/xmod_Opset16.onnx --no-check-certificate
12+
13+
onnxsim resnext26ts_Opset16.onnx ../resnet-test.onnx
14+
#onnxsim xmod_Opset16.onnx ../xmod-transformer-test.onnx
15+

0 commit comments

Comments
 (0)