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