Skip to content

Commit 4bb312a

Browse files
authored
[tools/onnx-subgraph] Add test model downloading scripts (#14690)
Add test onnx models downloading scripts for future smoking test ONE-DCO-1.0-Signed-off-by: Youxin Chen <yx113.chen@samsung.com>
1 parent 02e5d99 commit 4bb312a

File tree

2 files changed

+31
-0
lines changed

2 files changed

+31
-0
lines changed

tools/onnx_subgraph/CMakeLists.txt

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,3 +13,26 @@ 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_SUBGRAPH_FILES
18+
test_model_download.sh
19+
)
20+
foreach(ONNX_SUBGRAPH IN ITEMS ${ONNX_SUBGRAPH_FILES})
21+
set(ONNX_SUBGRAPH_FILE ${ONNX_SUBGRAPH})
22+
set(ONNX_SUBGRAPH_SRC "${CMAKE_CURRENT_SOURCE_DIR}/${ONNX_SUBGRAPH_FILE}")
23+
set(ONNX_SUBGRAPH_BIN "${CMAKE_CURRENT_BINARY_DIR}/scripts/${ONNX_SUBGRAPH_FILE}")
24+
set(ONNX_SUBGRAPH_TARGET "${ONNX_SUBGRAPH}_target")
25+
26+
add_custom_command(OUTPUT ${ONNX_SUBGRAPH_BIN}
27+
COMMAND ${CMAKE_COMMAND} -E copy "${ONNX_SUBGRAPH_SRC}" "${ONNX_SUBGRAPH_BIN}"
28+
DEPENDS ${ONNX_SUBGRAPH_SRC}
29+
COMMENT "Generate ${ONNX_SUBGRAPH_BIN}"
30+
)
31+
add_custom_target(${ONNX_SUBGRAPH_TARGET} ALL DEPENDS ${ONNX_SUBGRAPH_BIN})
32+
install(FILES ${ONNX_SUBGRAPH_BIN}
33+
PERMISSIONS OWNER_WRITE OWNER_READ OWNER_EXECUTE
34+
GROUP_READ GROUP_EXECUTE
35+
WORLD_READ WORLD_EXECUTE
36+
DESTINATION bin
37+
)
38+
endforeach(ONNX_SUBGRAPH)
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
#!/bin/bash
2+
3+
mkdir -p models
4+
5+
cd ./models
6+
wget https://media.githubusercontent.com/media/onnx/models/refs/heads/main/Computer_Vision/resnext26ts_Opset16_timm/resnext26ts_Opset16.onnx --no-check-certificate
7+
8+
onnxsim resnext26ts_Opset16.onnx ../resnet-test.onnx

0 commit comments

Comments
 (0)