diff --git a/tools/onnx_subgraph/CMakeLists.txt b/tools/onnx_subgraph/CMakeLists.txt index 29d8ab64d52..b6727d7f2e8 100644 --- a/tools/onnx_subgraph/CMakeLists.txt +++ b/tools/onnx_subgraph/CMakeLists.txt @@ -13,3 +13,26 @@ find_package(Python3 COMPONENTS Interpreter Development REQUIRED) include_directories(${CMAKE_CURRENT_BINARY_DIR}) include_directories(${Python3_INCLUDE_DIRS}) + +set(ONNX_SUBGRAPH_FILES + test_model_download.sh +) +foreach(ONNX_SUBGRAPH IN ITEMS ${ONNX_SUBGRAPH_FILES}) + set(ONNX_SUBGRAPH_FILE ${ONNX_SUBGRAPH}) + set(ONNX_SUBGRAPH_SRC "${CMAKE_CURRENT_SOURCE_DIR}/${ONNX_SUBGRAPH_FILE}") + set(ONNX_SUBGRAPH_BIN "${CMAKE_CURRENT_BINARY_DIR}/scripts/${ONNX_SUBGRAPH_FILE}") + set(ONNX_SUBGRAPH_TARGET "${ONNX_SUBGRAPH}_target") + + add_custom_command(OUTPUT ${ONNX_SUBGRAPH_BIN} + COMMAND ${CMAKE_COMMAND} -E copy "${ONNX_SUBGRAPH_SRC}" "${ONNX_SUBGRAPH_BIN}" + DEPENDS ${ONNX_SUBGRAPH_SRC} + COMMENT "Generate ${ONNX_SUBGRAPH_BIN}" + ) + add_custom_target(${ONNX_SUBGRAPH_TARGET} ALL DEPENDS ${ONNX_SUBGRAPH_BIN}) + install(FILES ${ONNX_SUBGRAPH_BIN} + PERMISSIONS OWNER_WRITE OWNER_READ OWNER_EXECUTE + GROUP_READ GROUP_EXECUTE + WORLD_READ WORLD_EXECUTE + DESTINATION bin + ) +endforeach(ONNX_SUBGRAPH) diff --git a/tools/onnx_subgraph/test_model_download.sh b/tools/onnx_subgraph/test_model_download.sh new file mode 100644 index 00000000000..fade6e7391d --- /dev/null +++ b/tools/onnx_subgraph/test_model_download.sh @@ -0,0 +1,8 @@ +#!/bin/bash + +mkdir -p models + +cd ./models +wget https://media.githubusercontent.com/media/onnx/models/refs/heads/main/Computer_Vision/resnext26ts_Opset16_timm/resnext26ts_Opset16.onnx --no-check-certificate + +onnxsim resnext26ts_Opset16.onnx ../resnet-test.onnx