File tree Expand file tree Collapse file tree 2 files changed +31
-0
lines changed
Expand file tree Collapse file tree 2 files changed +31
-0
lines changed Original file line number Diff line number Diff line change @@ -13,3 +13,26 @@ find_package(Python3 COMPONENTS Interpreter Development REQUIRED)
1313
1414include_directories (${CMAKE_CURRENT_BINARY_DIR} )
1515include_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 )
Original file line number Diff line number Diff line change 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
You can’t perform that action at this time.
0 commit comments