nihui
/
ncnn
Not watched
1
0
Fork 0
Code
Releases 46 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: a3a2548aa2
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'a3a2548aa2'
${ noResults }
ncnn/tools/tensorflow/CMakeLists.txt

25 lines
808 B
Raw Blame History 

  1. 

  2. find_package(Protobuf REQUIRED)

  3. 

  4. if(PROTOBUF_FOUND)

  5.     include_directories(${PROTOBUF_INCLUDE_DIR})

  6.     include_directories(${CMAKE_CURRENT_BINARY_DIR})

  7.     protobuf_generate_cpp(TENSORFLOW_PROTO_SRCS TENSORFLOW_PROTO_HDRS

  8.         attr_value.proto

  9.         function.proto

  10.         graph.proto

  11.         node_def.proto

  12.         op_def.proto

  13.         resource_handle.proto

  14.         tensor.proto

  15.         tensor_shape.proto

  16.         types.proto

  17.         versions.proto

  18.     )

  19.     add_executable(tensorflow2ncnn tensorflow2ncnn.cpp ${TENSORFLOW_PROTO_SRCS} ${TENSORFLOW_PROTO_HDRS})

  20.     set_target_properties(tensorflow2ncnn PROPERTIES CXX_STANDARD 11)

  21.     target_link_libraries(tensorflow2ncnn ${PROTOBUF_LIBRARIES})

  22. else()

  23.     message(WARNING "Protobuf not found, tensorflow model convert tool won't be built")

  24. endif()