2017-06-26 18:35:51 +08:00
if ( WINRT )
ocv_module_disable ( dnn )
endif ( )
2018-02-01 21:30:23 +08:00
if ( NOT HAVE_PROTOBUF )
2017-06-26 18:35:51 +08:00
ocv_module_disable ( opencv_dnn )
endif ( )
set ( the_description "Deep neural network module. It allows to load models from different frameworks and to make forward pass" )
2022-09-10 14:39:43 +08:00
ocv_add_dispatched_file_force_all ( "layers/layers_common" AVX AVX2 AVX512_SKX RVV LASX )
ocv_add_dispatched_file_force_all ( "int8layers/layers_common" AVX2 AVX512_SKX LASX )
2017-07-13 23:42:36 +08:00
2020-06-09 02:32:53 +08:00
ocv_add_module ( dnn opencv_core opencv_imgproc WRAP python java objc js )
2018-02-28 18:58:55 +08:00
2022-03-18 14:19:30 +08:00
include ( ${ CMAKE_CURRENT_LIST_DIR } /cmake/plugin.cmake )
2018-06-26 14:35:18 +08:00
ocv_option ( OPENCV_DNN_OPENCL "Build with OpenCL support" HAVE_OPENCL AND NOT APPLE )
2018-02-28 18:58:55 +08:00
if ( OPENCV_DNN_OPENCL AND HAVE_OPENCL )
2022-03-18 14:19:30 +08:00
ocv_target_compile_definitions ( ${ the_module } PRIVATE "CV_OCL4DNN=1" )
2018-02-28 18:58:55 +08:00
endif ( )
2021-11-24 05:15:31 +08:00
if ( WITH_WEBNN AND HAVE_WEBNN )
2022-03-18 14:19:30 +08:00
ocv_target_compile_definitions ( ${ the_module } PRIVATE "HAVE_WEBNN=1" )
2021-11-24 05:15:31 +08:00
endif ( )
2022-04-01 05:42:11 +08:00
if ( HAVE_TIMVX )
2022-03-18 14:19:30 +08:00
ocv_target_compile_definitions ( ${ the_module } PRIVATE "HAVE_TIMVX=1" )
2022-04-01 05:42:11 +08:00
endif ( )
2020-06-28 04:34:44 +08:00
ocv_option ( OPENCV_DNN_CUDA "Build with CUDA support"
H A V E _ C U D A
A N D H A V E _ C U B L A S
A N D H A V E _ C U D N N
)
2019-06-02 19:47:15 +08:00
2020-07-09 20:31:33 +08:00
if ( OPENCV_DNN_CUDA )
if ( HAVE_CUDA AND HAVE_CUBLAS AND HAVE_CUDNN )
2022-03-18 14:19:30 +08:00
ocv_target_compile_definitions ( ${ the_module } PRIVATE "CV_CUDA4DNN=1" )
2020-07-09 20:31:33 +08:00
else ( )
if ( NOT HAVE_CUDA )
message ( SEND_ERROR "DNN: CUDA backend requires CUDA Toolkit. Please resolve dependency or disable OPENCV_DNN_CUDA=OFF" )
elseif ( NOT HAVE_CUBLAS )
message ( SEND_ERROR "DNN: CUDA backend requires cuBLAS. Please resolve dependency or disable OPENCV_DNN_CUDA=OFF" )
elseif ( NOT HAVE_CUDNN )
message ( SEND_ERROR "DNN: CUDA backend requires cuDNN. Please resolve dependency or disable OPENCV_DNN_CUDA=OFF" )
endif ( )
endif ( )
2019-06-02 19:47:15 +08:00
endif ( )
2022-03-18 14:19:30 +08:00
2019-06-02 19:47:15 +08:00
ocv_cmake_hook_append ( INIT_MODULE_SOURCES_opencv_dnn "${CMAKE_CURRENT_LIST_DIR}/cmake/hooks/INIT_MODULE_SOURCES_opencv_dnn.cmake" )
2022-03-18 14:19:30 +08:00
2021-11-04 17:59:19 +08:00
if ( HAVE_TENGINE )
2022-03-18 14:19:30 +08:00
ocv_target_compile_definitions ( ${ the_module } PRIVATE "HAVE_TENGINE=1" )
2021-11-04 17:59:19 +08:00
endif ( )
2022-03-18 14:19:30 +08:00
2017-06-26 18:35:51 +08:00
if ( MSVC )
add_definitions ( -D_CRT_SECURE_NO_WARNINGS=1 )
ocv_warnings_disable ( CMAKE_CXX_FLAGS /wd4244 /wd4267 /wd4018 /wd4355 /wd4800 /wd4251 /wd4996 /wd4146
/ w d 4 3 0 5 / w d 4 1 2 7 / w d 4 1 0 0 / w d 4 5 1 2 / w d 4 1 2 5 / w d 4 3 8 9 / w d 4 5 1 0 / w d 4 6 1 0
/ w d 4 7 0 2 / w d 4 4 5 6 / w d 4 4 5 7 / w d 4 0 6 5 / w d 4 3 1 0 / w d 4 6 6 1 / w d 4 5 0 6
)
2021-11-30 12:27:39 +08:00
if ( MSVC_VERSION LESS 1920 ) # MSVS 2015/2017, .pb.cc generated files
2021-11-10 20:03:45 +08:00
ocv_warnings_disable ( CMAKE_CXX_FLAGS /wd4309 ) # 'static_cast': truncation of constant value
endif ( )
if ( MSVC_VERSION LESS 1920 ) # <MSVS2019, .pb.cc generated files
ocv_warnings_disable ( CMAKE_CXX_FLAGS /wd4189 ) # local variable is initialized but not referenced
ocv_warnings_disable ( CMAKE_CXX_FLAGS /wd4592 ) # symbol will be dynamically initialized (implementation limitation)
endif ( )
2017-06-26 18:35:51 +08:00
else ( )
ocv_warnings_disable ( CMAKE_CXX_FLAGS -Wno-deprecated -Wmissing-prototypes -Wmissing-declarations -Wshadow
2018-11-15 04:25:23 +08:00
- W u n u s e d - p a r a m e t e r - W s i g n - c o m p a r e
2017-06-26 18:35:51 +08:00
)
endif ( )
2019-08-08 20:49:32 +08:00
if ( HAVE_CUDA )
ocv_warnings_disable ( CMAKE_CXX_FLAGS -Wundef )
endif ( )
2018-11-15 04:25:23 +08:00
if ( NOT HAVE_CXX11 )
ocv_warnings_disable ( CMAKE_CXX_FLAGS -Wno-undef ) # LANG_CXX11 from protobuf files
endif ( )
2017-06-26 18:35:51 +08:00
if ( APPLE_FRAMEWORK )
ocv_warnings_disable ( CMAKE_CXX_FLAGS -Wshorten-64-to-32 )
endif ( )
if ( ANDROID )
add_definitions ( -DDISABLE_POSIX_MEMALIGN -DTH_DISABLE_HEAP_TRACKING )
endif ( )
2018-07-09 22:19:35 +08:00
if ( NOT BUILD_PROTOBUF )
2022-03-18 14:19:30 +08:00
ocv_target_compile_definitions ( ${ the_module } PRIVATE "OPENCV_DNN_EXTERNAL_PROTOBUF=1" )
2018-07-09 22:19:35 +08:00
endif ( )
2022-03-18 14:19:30 +08:00
ocv_target_compile_definitions ( ${ the_module } PRIVATE "HAVE_PROTOBUF=1" )
2018-02-01 21:30:23 +08:00
2018-02-12 20:07:39 +08:00
#suppress warnings in autogenerated caffe.pb.* files
2017-06-26 18:35:51 +08:00
ocv_warnings_disable ( CMAKE_CXX_FLAGS
/ w d 4 1 2 5 / w d 4 2 6 7 / w d 4 1 2 7 / w d 4 2 4 4 / w d 4 5 1 2 / w d 4 7 0 2
/ w d 4 4 5 6 / w d 4 5 1 0 / w d 4 6 1 0 / w d 4 8 0 0
2018-01-16 03:49:36 +08:00
/ w d 4 7 0 1 / w d 4 7 0 3 # potentially uninitialized local/pointer variable 'value' used
/ w d 4 5 0 5 # unreferenced local function has been removed
2019-12-12 01:16:58 +08:00
/ w d 4 4 5 8 # declaration of 'x' hides class member. GCC still works, MSVC bug is here: https://developercommunity.visualstudio.com/content/problem/219311/c-c4458-declaration-hides-class-member-warning-iss.html
2017-06-26 18:35:51 +08:00
- w d 8 5 8 - w d 2 1 9 6
2018-01-16 03:49:36 +08:00
- W i n v a l i d - o f f s e t o f # Apple Clang (attr_value.pb.cc)
2017-06-26 18:35:51 +08:00
)
2020-04-26 12:42:42 +08:00
set ( include_dirs "" )
set ( libs "" )
2017-06-26 18:35:51 +08:00
if ( PROTOBUF_UPDATE_FILES )
2018-09-11 02:07:51 +08:00
file ( GLOB proto_files "${CMAKE_CURRENT_LIST_DIR}/src/tensorflow/*.proto" "${CMAKE_CURRENT_LIST_DIR}/src/caffe/opencv-caffe.proto" "${CMAKE_CURRENT_LIST_DIR}/src/onnx/opencv-onnx.proto" )
2017-11-15 19:07:41 +08:00
set ( PROTOBUF_GENERATE_CPP_APPEND_PATH ON ) # required for tensorflow
2018-02-01 21:30:23 +08:00
protobuf_generate_cpp ( fw_srcs fw_hdrs ${ proto_files } )
2017-06-26 18:35:51 +08:00
else ( )
2018-09-11 02:07:51 +08:00
file ( GLOB fw_srcs "${CMAKE_CURRENT_LIST_DIR}/misc/tensorflow/*.cc" "${CMAKE_CURRENT_LIST_DIR}/misc/caffe/opencv-caffe.pb.cc" "${CMAKE_CURRENT_LIST_DIR}/misc/onnx/opencv-onnx.pb.cc" )
file ( GLOB fw_hdrs "${CMAKE_CURRENT_LIST_DIR}/misc/tensorflow/*.h" "${CMAKE_CURRENT_LIST_DIR}/misc/caffe/opencv-caffe.pb.h" "${CMAKE_CURRENT_LIST_DIR}/misc/onnx/opencv-onnx.pb.h" )
set ( fw_inc "${CMAKE_CURRENT_LIST_DIR}/misc/caffe" "${CMAKE_CURRENT_LIST_DIR}/misc/tensorflow" "${CMAKE_CURRENT_LIST_DIR}/misc/onnx" )
2017-06-26 18:35:51 +08:00
endif ( )
2020-04-26 12:42:42 +08:00
list ( APPEND include_dirs ${ fw_inc } )
list ( APPEND libs ${ Protobuf_LIBRARIES } )
if ( NOT BUILD_PROTOBUF )
list ( APPEND include_dirs ${ Protobuf_INCLUDE_DIRS } )
endif ( )
2019-12-02 21:16:06 +08:00
2018-02-28 18:58:55 +08:00
set ( sources_options "" )
2019-06-02 19:47:15 +08:00
2020-04-26 12:42:42 +08:00
list ( APPEND libs ${ LAPACK_LIBRARIES } )
2018-02-28 18:58:55 +08:00
if ( OPENCV_DNN_OPENCL AND HAVE_OPENCL )
list ( APPEND include_dirs ${ OPENCL_INCLUDE_DIRS } )
else ( )
set ( sources_options EXCLUDE_OPENCL )
endif ( )
2018-03-28 21:34:37 +08:00
2019-06-02 19:47:15 +08:00
if ( OPENCV_DNN_CUDA AND HAVE_CUDA AND HAVE_CUBLAS AND HAVE_CUDNN )
list ( APPEND include_dirs ${ CUDA_TOOLKIT_INCLUDE } ${ CUDNN_INCLUDE_DIRS } )
Merge pull request #14827 from YashasSamaga:cuda4dnn-csl-low
CUDA backend for the DNN module
* stub cuda4dnn design
* minor fixes for tests and doxygen
* add csl public api directory to module headers
* add low-level CSL components
* add high-level CSL components
* integrate csl::Tensor into backbone code
* switch to CPU iff unsupported; otherwise, fail on error
* add fully connected layer
* add softmax layer
* add activation layers
* support arbitary rank TensorDescriptor
* pass input wrappers to `initCUDA()`
* add 1d/2d/3d-convolution
* add pooling layer
* reorganize and refactor code
* fixes for gcc, clang and doxygen; remove cxx14/17 code
* add blank_layer
* add LRN layer
* add rounding modes for pooling layer
* split tensor.hpp into tensor.hpp and tensor_ops.hpp
* add concat layer
* add scale layer
* add batch normalization layer
* split math.cu into activations.cu and math.hpp
* add eltwise layer
* add flatten layer
* add tensor transform api
* add asymmetric padding support for convolution layer
* add reshape layer
* fix rebase issues
* add permute layer
* add padding support for concat layer
* refactor and reorganize code
* add normalize layer
* optimize bias addition in scale layer
* add prior box layer
* fix and optimize normalize layer
* add asymmetric padding support for pooling layer
* add event API
* improve pooling performance for some padding scenarios
* avoid over-allocation of compute resources to kernels
* improve prior box performance
* enable layer fusion
* add const layer
* add resize layer
* add slice layer
* add padding layer
* add deconvolution layer
* fix channelwise ReLU initialization
* add vector traits
* add vectorized versions of relu, clipped_relu, power
* add vectorized concat kernels
* improve concat_with_offsets performance
* vectorize scale and bias kernels
* add support for multi-billion element tensors
* vectorize prior box kernels
* fix address alignment check
* improve bias addition performance of conv/deconv/fc layers
* restructure code for supporting multiple targets
* add DNN_TARGET_CUDA_FP64
* add DNN_TARGET_FP16
* improve vectorization
* add region layer
* improve tensor API, add dynamic ranks
1. use ManagedPtr instead of a Tensor in backend wrapper
2. add new methods to tensor classes
- size_range: computes the combined size of for a given axis range
- tensor span/view can be constructed from a raw pointer and shape
3. the tensor classes can change their rank at runtime (previously rank was fixed at compile-time)
4. remove device code from tensor classes (as they are unused)
5. enforce strict conditions on tensor class APIs to improve debugging ability
* fix parametric relu activation
* add squeeze/unsqueeze tensor API
* add reorg layer
* optimize permute and enable 2d permute
* enable 1d and 2d slice
* add split layer
* add shuffle channel layer
* allow tensors of different ranks in reshape primitive
* patch SliceOp to allow Crop Layer
* allow extra shape inputs in reshape layer
* use `std::move_backward` instead of `std::move` for insert in resizable_static_array
* improve workspace management
* add spatial LRN
* add nms (cpu) to region layer
* add max pooling with argmax ( and a fix to limits.hpp)
* add max unpooling layer
* rename DNN_TARGET_CUDA_FP32 to DNN_TARGET_CUDA
* update supportBackend to be more rigorous
* remove stray include from preventing non-cuda build
* include op_cuda.hpp outside condition #if
* refactoring, fixes and many optimizations
* drop DNN_TARGET_CUDA_FP64
* fix gcc errors
* increase max. tensor rank limit to six
* add Interp layer
* drop custom layers; use BackendNode
* vectorize activation kernels
* fixes for gcc
* remove wrong assertion
* fix broken assertion in unpooling primitive
* fix build errors in non-CUDA build
* completely remove workspace from public API
* fix permute layer
* enable accuracy and perf. tests for DNN_TARGET_CUDA
* add asynchronous forward
* vectorize eltwise ops
* vectorize fill kernel
* fixes for gcc
* remove CSL headers from public API
* remove csl header source group from cmake
* update min. cudnn version in cmake
* add numerically stable FP32 log1pexp
* refactor code
* add FP16 specialization to cudnn based tensor addition
* vectorize scale1 and bias1 + minor refactoring
* fix doxygen build
* fix invalid alignment assertion
* clear backend wrappers before allocateLayers
* ignore memory lock failures
* do not allocate internal blobs
* integrate NVTX
* add numerically stable half precision log1pexp
* fix indentation, following coding style, improve docs
* remove accidental modification of IE code
* Revert "add asynchronous forward"
This reverts commit 1154b9da9da07e9b52f8a81bdcea48cf31c56f70.
* [cmake] throw error for unsupported CC versions
* fix rebase issues
* add more docs, refactor code, fix bugs
* minor refactoring and fixes
* resolve warnings/errors from clang
* remove haveCUDA() checks from supportBackend()
* remove NVTX integration
* changes based on review comments
* avoid exception when no CUDA device is present
* add color code for CUDA in Net::dump
2019-10-21 19:28:00 +08:00
set ( CC_LIST ${ CUDA_ARCH_BIN } )
separate_arguments ( CC_LIST )
foreach ( cc ${ CC_LIST } )
2020-01-15 23:28:37 +08:00
if ( cc VERSION_LESS 3.0 )
message ( FATAL_ERROR "CUDA backend for DNN module requires CC 3.0 or higher. Please remove unsupported architectures from CUDA_ARCH_BIN option or disable OPENCV_DNN_CUDA=OFF." )
Merge pull request #14827 from YashasSamaga:cuda4dnn-csl-low
CUDA backend for the DNN module
* stub cuda4dnn design
* minor fixes for tests and doxygen
* add csl public api directory to module headers
* add low-level CSL components
* add high-level CSL components
* integrate csl::Tensor into backbone code
* switch to CPU iff unsupported; otherwise, fail on error
* add fully connected layer
* add softmax layer
* add activation layers
* support arbitary rank TensorDescriptor
* pass input wrappers to `initCUDA()`
* add 1d/2d/3d-convolution
* add pooling layer
* reorganize and refactor code
* fixes for gcc, clang and doxygen; remove cxx14/17 code
* add blank_layer
* add LRN layer
* add rounding modes for pooling layer
* split tensor.hpp into tensor.hpp and tensor_ops.hpp
* add concat layer
* add scale layer
* add batch normalization layer
* split math.cu into activations.cu and math.hpp
* add eltwise layer
* add flatten layer
* add tensor transform api
* add asymmetric padding support for convolution layer
* add reshape layer
* fix rebase issues
* add permute layer
* add padding support for concat layer
* refactor and reorganize code
* add normalize layer
* optimize bias addition in scale layer
* add prior box layer
* fix and optimize normalize layer
* add asymmetric padding support for pooling layer
* add event API
* improve pooling performance for some padding scenarios
* avoid over-allocation of compute resources to kernels
* improve prior box performance
* enable layer fusion
* add const layer
* add resize layer
* add slice layer
* add padding layer
* add deconvolution layer
* fix channelwise ReLU initialization
* add vector traits
* add vectorized versions of relu, clipped_relu, power
* add vectorized concat kernels
* improve concat_with_offsets performance
* vectorize scale and bias kernels
* add support for multi-billion element tensors
* vectorize prior box kernels
* fix address alignment check
* improve bias addition performance of conv/deconv/fc layers
* restructure code for supporting multiple targets
* add DNN_TARGET_CUDA_FP64
* add DNN_TARGET_FP16
* improve vectorization
* add region layer
* improve tensor API, add dynamic ranks
1. use ManagedPtr instead of a Tensor in backend wrapper
2. add new methods to tensor classes
- size_range: computes the combined size of for a given axis range
- tensor span/view can be constructed from a raw pointer and shape
3. the tensor classes can change their rank at runtime (previously rank was fixed at compile-time)
4. remove device code from tensor classes (as they are unused)
5. enforce strict conditions on tensor class APIs to improve debugging ability
* fix parametric relu activation
* add squeeze/unsqueeze tensor API
* add reorg layer
* optimize permute and enable 2d permute
* enable 1d and 2d slice
* add split layer
* add shuffle channel layer
* allow tensors of different ranks in reshape primitive
* patch SliceOp to allow Crop Layer
* allow extra shape inputs in reshape layer
* use `std::move_backward` instead of `std::move` for insert in resizable_static_array
* improve workspace management
* add spatial LRN
* add nms (cpu) to region layer
* add max pooling with argmax ( and a fix to limits.hpp)
* add max unpooling layer
* rename DNN_TARGET_CUDA_FP32 to DNN_TARGET_CUDA
* update supportBackend to be more rigorous
* remove stray include from preventing non-cuda build
* include op_cuda.hpp outside condition #if
* refactoring, fixes and many optimizations
* drop DNN_TARGET_CUDA_FP64
* fix gcc errors
* increase max. tensor rank limit to six
* add Interp layer
* drop custom layers; use BackendNode
* vectorize activation kernels
* fixes for gcc
* remove wrong assertion
* fix broken assertion in unpooling primitive
* fix build errors in non-CUDA build
* completely remove workspace from public API
* fix permute layer
* enable accuracy and perf. tests for DNN_TARGET_CUDA
* add asynchronous forward
* vectorize eltwise ops
* vectorize fill kernel
* fixes for gcc
* remove CSL headers from public API
* remove csl header source group from cmake
* update min. cudnn version in cmake
* add numerically stable FP32 log1pexp
* refactor code
* add FP16 specialization to cudnn based tensor addition
* vectorize scale1 and bias1 + minor refactoring
* fix doxygen build
* fix invalid alignment assertion
* clear backend wrappers before allocateLayers
* ignore memory lock failures
* do not allocate internal blobs
* integrate NVTX
* add numerically stable half precision log1pexp
* fix indentation, following coding style, improve docs
* remove accidental modification of IE code
* Revert "add asynchronous forward"
This reverts commit 1154b9da9da07e9b52f8a81bdcea48cf31c56f70.
* [cmake] throw error for unsupported CC versions
* fix rebase issues
* add more docs, refactor code, fix bugs
* minor refactoring and fixes
* resolve warnings/errors from clang
* remove haveCUDA() checks from supportBackend()
* remove NVTX integration
* changes based on review comments
* avoid exception when no CUDA device is present
* add color code for CUDA in Net::dump
2019-10-21 19:28:00 +08:00
endif ( )
endforeach ( )
unset ( CC_LIST )
2019-06-02 19:47:15 +08:00
else ( )
set ( sources_options ${ sources_options } EXCLUDE_CUDA )
endif ( )
2020-03-09 22:59:23 +08:00
if ( HAVE_TENGINE )
list ( APPEND include_dirs ${ TENGINE_INCLUDE_DIRS } )
2020-09-23 17:34:29 +08:00
list ( APPEND libs -Wl,--whole-archive ${ TENGINE_LIBRARIES } -Wl,--no-whole-archive )
2020-03-09 22:59:23 +08:00
endif ( )
2022-04-01 05:42:11 +08:00
if ( HAVE_TIMVX )
list ( APPEND include_dirs ${ TIMVX_INCLUDE_DIR } )
list ( APPEND libs -Wl,--whole-archive ${ TIMVX_LIBRARY } -Wl,--no-whole-archive )
endif ( )
2021-11-24 05:15:31 +08:00
set ( webnn_srcs "" )
if ( NOT EMSCRIPTEN )
if ( HAVE_WEBNN )
list ( APPEND include_dirs ${ WEBNN_HEADER_DIRS } )
list ( APPEND include_dirs ${ WEBNN_INCLUDE_DIRS } )
list ( APPEND libs -Wl,--whole-archive ${ WEBNN_LIBRARIES } -Wl,--no-whole-archive )
list ( APPEND webnn_srcs $ENV{ WEBNN_NATIVE_DIR } /gen/src/webnn/webnn_cpp.cpp )
endif ( )
endif ( )
2018-02-28 18:58:55 +08:00
ocv_module_include_directories ( ${ include_dirs } )
2018-03-20 18:26:13 +08:00
if ( CMAKE_CXX_COMPILER_ID STREQUAL "GNU" )
ocv_append_source_files_cxx_compiler_options ( fw_srcs "-Wno-suggest-override" ) # GCC
2020-03-27 03:05:07 +08:00
ocv_append_source_files_cxx_compiler_options ( fw_srcs "-Wno-array-bounds" ) # GCC 9.3.0 (Ubuntu 20.04)
2018-03-20 18:26:13 +08:00
elseif ( CMAKE_CXX_COMPILER_ID STREQUAL "Clang" )
ocv_append_source_files_cxx_compiler_options ( fw_srcs "-Wno-inconsistent-missing-override" ) # Clang
endif ( )
2019-12-02 21:16:06 +08:00
set ( dnn_runtime_libs "" )
2022-02-09 22:25:08 +08:00
2022-03-18 14:19:30 +08:00
file ( GLOB_RECURSE dnn_srcs
" $ { C M A K E _ C U R R E N T _ L I S T _ D I R } / s r c / * . c p p "
)
file ( GLOB_RECURSE dnn_int_hdrs
" $ { C M A K E _ C U R R E N T _ L I S T _ D I R } / s r c / * . h p p "
" $ { C M A K E _ C U R R E N T _ L I S T _ D I R } / s r c / * . h "
)
set ( dnn_plugin_srcs ${ dnn_srcs } ${ dnn_int_hdrs } )
ocv_list_filterout_ex ( dnn_plugin_srcs
" / s r c / d n n . c p p $ | / s r c / d n n _ u t i l s . c p p $ | / s r c / d n n _ u t i l s . c p p $ | / s r c / d n n _ r e a d . c p p $ | / s r c / r e g i s t r y . c p p $ | / s r c / b a c k e n d . c p p $ "
# importers
" / s r c / ( c a f f e | d a r k n e t | o n n x | t e n s o r f l o w | t o r c h ) / "
# executors
" / s r c / ( c u d a | c u d a 4 d n n | o c l 4 d n n | v k c o m | w e b n n ) / "
)
2022-02-09 22:25:08 +08:00
ocv_option ( OPENCV_DNN_OPENVINO "Build with OpenVINO support (2021.4+)" ( TARGET ocv.3rdparty.openvino ) )
if ( TARGET ocv.3rdparty.openvino AND OPENCV_DNN_OPENVINO )
if ( NOT HAVE_OPENVINO AND NOT HAVE_NGRAPH )
message ( FATAL_ERROR "DNN: Inference Engine is not supported without enabled 'nGraph'. Check build configuration." )
endif ( )
2022-03-18 14:19:30 +08:00
if ( "openvino" IN_LIST DNN_PLUGIN_LIST OR DNN_PLUGIN_LIST STREQUAL "all" )
# plugin doesn't support PCH, separate directory scope is necessary
# opencv_world requires absolute path
add_subdirectory ( "${CMAKE_CURRENT_LIST_DIR}/misc/plugin/openvino" "${CMAKE_CURRENT_BINARY_DIR}/dnn_plugin_openvino" )
elseif ( NOT OPENCV_DNN_BUILTIN_BACKEND )
list ( APPEND dnn_runtime_libs ocv.3rdparty.openvino )
endif ( )
2019-12-02 21:16:06 +08:00
endif ( )
2021-11-24 05:15:31 +08:00
ocv_glob_module_sources ( ${ sources_options } SOURCES ${ fw_srcs } ${ webnn_srcs } )
2019-12-02 21:16:06 +08:00
ocv_create_module ( ${ libs } ${ dnn_runtime_libs } )
2017-06-26 18:35:51 +08:00
ocv_add_samples ( )
2019-12-02 21:16:06 +08:00
ocv_add_accuracy_tests ( ${ dnn_runtime_libs } )
2019-03-29 21:42:58 +08:00
set ( perf_path "${CMAKE_CURRENT_LIST_DIR}/perf" )
file ( GLOB_RECURSE perf_srcs "${perf_path}/*.cpp" )
file ( GLOB_RECURSE perf_hdrs "${perf_path}/*.hpp" "${perf_path}/*.h" )
2022-01-15 05:53:19 +08:00
ocv_add_perf_tests ( ${ dnn_runtime_libs }
2019-03-31 16:51:09 +08:00
F I L E S t e s t _ c o m m o n " $ { C M A K E _ C U R R E N T _ L I S T _ D I R } / t e s t / t e s t _ c o m m o n . h p p " " $ { C M A K E _ C U R R E N T _ L I S T _ D I R } / t e s t / t e s t _ c o m m o n . i m p l . h p p "
2019-03-29 21:42:58 +08:00
F I L E S S r c $ { p e r f _ s r c s }
F I L E S I n c l u d e $ { p e r f _ h d r s }
)
2017-06-26 18:35:51 +08:00
2021-04-26 23:36:51 +08:00
ocv_option ( OPENCV_DNN_PERF_CAFFE "Add performance tests of Caffe framework" OFF )
ocv_option ( OPENCV_DNN_PERF_CLCAFFE "Add performance tests of clCaffe framework" OFF )
2017-08-04 20:09:57 +08:00
if ( BUILD_PERF_TESTS )
2021-04-26 23:36:51 +08:00
if ( OPENCV_DNN_PERF_CAFFE
O R $ { t h e _ m o d u l e } _ P E R F _ C A F F E # compatibility for deprecated option
)
2017-08-04 20:09:57 +08:00
find_package ( Caffe QUIET )
if ( Caffe_FOUND )
2022-03-18 14:19:30 +08:00
ocv_target_compile_definitions ( opencv_perf_dnn PRIVATE "HAVE_CAFFE=1" )
2017-08-04 20:09:57 +08:00
ocv_target_link_libraries ( opencv_perf_dnn caffe )
endif ( )
2021-04-26 23:36:51 +08:00
elseif ( OPENCV_DNN_PERF_CLCAFFE
O R $ { t h e _ m o d u l e } _ P E R F _ C A F F E # compatibility for deprecated option
)
2017-08-04 20:09:57 +08:00
find_package ( Caffe QUIET )
if ( Caffe_FOUND )
2022-03-18 14:19:30 +08:00
ocv_target_compile_definitions ( opencv_perf_dnn PRIVATE "HAVE_CLCAFFE=1" )
2017-08-04 20:09:57 +08:00
ocv_target_link_libraries ( opencv_perf_dnn caffe )
endif ( )
endif ( )
endif ( )
2022-03-18 14:19:30 +08:00
if ( DNN_ENABLE_PLUGINS )
ocv_target_compile_definitions ( ${ the_module } PRIVATE ENABLE_PLUGINS )
if ( TARGET opencv_test_dnn )
ocv_target_compile_definitions ( opencv_test_dnn PRIVATE ENABLE_PLUGINS )
endif ( )
if ( OPENCV_DEBUG_POSTFIX )
ocv_append_source_file_compile_definitions ( "${CMAKE_CURRENT_LIST_DIR}/src/backend.cpp" "DEBUG_POSTFIX=${OPENCV_DEBUG_POSTFIX}" )
endif ( )
endif ( )
ocv_option ( OPENCV_TEST_DNN_OPENVINO "Build test with OpenVINO code" ( TARGET ocv.3rdparty.openvino ) )
if ( TARGET ocv.3rdparty.openvino AND OPENCV_TEST_DNN_OPENVINO )
if ( TARGET opencv_test_dnn )
ocv_target_link_libraries ( opencv_test_dnn ocv.3rdparty.openvino )
endif ( )
endif ( )