mirror of
https://github.com/opencv/opencv.git
synced 2024-11-27 12:40:05 +08:00
Merge pull request #23604 from dkurt:dnn_no_protobuf
Build DNN without Protobuf DNN module can be built without Protobuf for Darknet, TFLite, OpenVINO, Torch (not PyTorch) models. ``` cmake \ -DCMAKE_BUILD_TYPE=Release \ -DBUILD_LIST=dnn \ -DWITH_PROTOBUF=OFF \ -DWITH_OPENCL=OFF 7.1M lib/libopencv_dnn.so.4.7.0 ``` ``` cmake \ -DCMAKE_BUILD_TYPE=Release \ -DBUILD_LIST=dnn \ -DWITH_OPENCL=OFF 3.9M lib/libopencv_dnn.so.4.7.0 ``` ### Pull Request Readiness Checklist See details at https://github.com/opencv/opencv/wiki/How_to_contribute#making-a-good-pull-request - [x] I agree to contribute to the project under Apache 2 License. - [x] To the best of my knowledge, the proposed patch is not based on a code under GPL or another license that is incompatible with OpenCV - [x] The PR is proposed to the proper branch - [x] There is a reference to the original bug report and related work - [x] There is accuracy test, performance test and test data in opencv_extra repository, if applicable Patch to opencv_extra has the same branch name. - [x] The feature is well documented and sample code can be built with the project CMake
This commit is contained in:
parent
85b04f0b4d
commit
a8d3d1f6f9
@ -2,10 +2,6 @@ if(WINRT)
|
||||
ocv_module_disable(dnn)
|
||||
endif()
|
||||
|
||||
if(NOT HAVE_PROTOBUF)
|
||||
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")
|
||||
|
||||
ocv_add_dispatched_file_force_all("layers/layers_common" AVX AVX2 AVX512_SKX RVV LASX)
|
||||
@ -104,8 +100,6 @@ if(NOT BUILD_PROTOBUF)
|
||||
ocv_target_compile_definitions(${the_module} PRIVATE "OPENCV_DNN_EXTERNAL_PROTOBUF=1")
|
||||
endif()
|
||||
|
||||
ocv_target_compile_definitions(${the_module} PRIVATE "HAVE_PROTOBUF=1")
|
||||
|
||||
#suppress warnings in autogenerated caffe.pb.* files
|
||||
ocv_warnings_disable(CMAKE_CXX_FLAGS
|
||||
/wd4125 /wd4267 /wd4127 /wd4244 /wd4512 /wd4702
|
||||
@ -120,14 +114,18 @@ ocv_warnings_disable(CMAKE_CXX_FLAGS
|
||||
set(include_dirs "")
|
||||
set(libs "")
|
||||
|
||||
if(PROTOBUF_UPDATE_FILES)
|
||||
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")
|
||||
set(PROTOBUF_GENERATE_CPP_APPEND_PATH ON) # required for tensorflow
|
||||
protobuf_generate_cpp(fw_srcs fw_hdrs ${proto_files})
|
||||
else()
|
||||
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")
|
||||
if(HAVE_PROTOBUF)
|
||||
ocv_target_compile_definitions(${the_module} PRIVATE "HAVE_PROTOBUF=1")
|
||||
|
||||
if(PROTOBUF_UPDATE_FILES)
|
||||
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")
|
||||
set(PROTOBUF_GENERATE_CPP_APPEND_PATH ON) # required for tensorflow
|
||||
protobuf_generate_cpp(fw_srcs fw_hdrs ${proto_files})
|
||||
else()
|
||||
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")
|
||||
endif()
|
||||
endif()
|
||||
|
||||
ocv_option(OPENCV_DNN_TFLITE "Build with TFLite support" (TARGET ocv.3rdparty.flatbuffers))
|
||||
|
@ -591,7 +591,23 @@ Net readNetFromCaffe(const std::vector<uchar>& bufferProto, const std::vector<uc
|
||||
bufferModelPtr, bufferModel.size());
|
||||
}
|
||||
|
||||
#endif //HAVE_PROTOBUF
|
||||
#else // HAVE_PROTOBUF
|
||||
|
||||
#define DNN_PROTOBUF_UNSUPPORTED() CV_Error(Error::StsError, "DNN/Caffe: Build OpenCV with Protobuf to import Caffe models")
|
||||
|
||||
Net readNetFromCaffe(const String &, const String &) {
|
||||
DNN_PROTOBUF_UNSUPPORTED();
|
||||
}
|
||||
|
||||
Net readNetFromCaffe(const char *, size_t, const char *, size_t) {
|
||||
DNN_PROTOBUF_UNSUPPORTED();
|
||||
}
|
||||
|
||||
Net readNetFromCaffe(const std::vector<uchar>&, const std::vector<uchar>&) {
|
||||
DNN_PROTOBUF_UNSUPPORTED();
|
||||
}
|
||||
|
||||
#endif // HAVE_PROTOBUF
|
||||
|
||||
CV__DNN_INLINE_NS_END
|
||||
}} // namespace
|
||||
|
@ -42,7 +42,7 @@
|
||||
#include "precomp.hpp"
|
||||
#include <opencv2/dnn/layer.details.hpp>
|
||||
|
||||
#if !defined(BUILD_PLUGIN)
|
||||
#if defined(HAVE_PROTOBUF) && !defined(BUILD_PLUGIN)
|
||||
#include <google/protobuf/stubs/common.h>
|
||||
#endif
|
||||
|
||||
@ -60,7 +60,7 @@ Mutex& getInitializationMutex()
|
||||
// force initialization (single-threaded environment)
|
||||
Mutex* __initialization_mutex_initializer = &getInitializationMutex();
|
||||
|
||||
#if !defined(BUILD_PLUGIN)
|
||||
#if defined(HAVE_PROTOBUF) && !defined(BUILD_PLUGIN)
|
||||
namespace {
|
||||
using namespace google::protobuf;
|
||||
class ProtobufShutdown {
|
||||
@ -80,7 +80,7 @@ void initializeLayerFactory()
|
||||
{
|
||||
CV_TRACE_FUNCTION();
|
||||
|
||||
#if !defined(BUILD_PLUGIN)
|
||||
#if defined(HAVE_PROTOBUF) && !defined(BUILD_PLUGIN)
|
||||
static ProtobufShutdown protobufShutdown; CV_UNUSED(protobufShutdown);
|
||||
#endif
|
||||
|
||||
|
@ -7,6 +7,7 @@
|
||||
|
||||
#include "../precomp.hpp"
|
||||
|
||||
#ifdef HAVE_PROTOBUF
|
||||
#include "../graph_simplifier.hpp"
|
||||
#include "onnx_graph_simplifier.hpp"
|
||||
|
||||
@ -1265,3 +1266,4 @@ Mat getMatFromTensor(const opencv_onnx::TensorProto& tensor_proto)
|
||||
|
||||
CV__DNN_INLINE_NS_END
|
||||
}} // namespace cv::dnn
|
||||
#endif // HAVE_PROTOBUF
|
||||
|
@ -7,6 +7,7 @@
|
||||
|
||||
#ifndef __OPENCV_DNN_ONNX_SIMPLIFIER_HPP__
|
||||
#define __OPENCV_DNN_ONNX_SIMPLIFIER_HPP__
|
||||
#ifdef HAVE_PROTOBUF
|
||||
|
||||
#if defined(__GNUC__) && __GNUC__ >= 5
|
||||
#pragma GCC diagnostic push
|
||||
@ -36,4 +37,5 @@ Mat getMatFromTensor(const opencv_onnx::TensorProto& tensor_proto);
|
||||
CV__DNN_INLINE_NS_END
|
||||
}} // namespace dnn, namespace cv
|
||||
|
||||
#endif // HAVE_PROTOBUF
|
||||
#endif // __OPENCV_DNN_ONNX_SIMPLIFIER_HPP__
|
||||
|
@ -43,6 +43,7 @@
|
||||
#endif
|
||||
|
||||
#include "onnx_graph_simplifier.hpp"
|
||||
#endif
|
||||
|
||||
namespace cv {
|
||||
namespace dnn {
|
||||
@ -50,6 +51,7 @@ CV__DNN_INLINE_NS_BEGIN
|
||||
|
||||
extern bool DNN_DIAGNOSTICS_RUN;
|
||||
|
||||
#ifdef HAVE_PROTOBUF
|
||||
class ONNXLayerHandler;
|
||||
|
||||
template <typename T>
|
||||
@ -4170,7 +4172,27 @@ Mat readTensorFromONNX(const String& path)
|
||||
return mat;
|
||||
}
|
||||
|
||||
#else // HAVE_PROTOBUF
|
||||
|
||||
#define DNN_PROTOBUF_UNSUPPORTED() CV_Error(Error::StsError, "DNN/ONNX: Build OpenCV with Protobuf to import ONNX models")
|
||||
|
||||
Net readNetFromONNX(const String&) {
|
||||
DNN_PROTOBUF_UNSUPPORTED();
|
||||
}
|
||||
|
||||
Net readNetFromONNX(const char*, size_t) {
|
||||
DNN_PROTOBUF_UNSUPPORTED();
|
||||
}
|
||||
|
||||
Net readNetFromONNX(const std::vector<uchar>&) {
|
||||
DNN_PROTOBUF_UNSUPPORTED();
|
||||
}
|
||||
|
||||
Mat readTensorFromONNX(const String&) {
|
||||
DNN_PROTOBUF_UNSUPPORTED();
|
||||
}
|
||||
|
||||
#endif // HAVE_PROTOBUF
|
||||
|
||||
CV__DNN_INLINE_NS_END
|
||||
}} // namespace
|
||||
|
||||
#endif
|
||||
|
@ -36,7 +36,7 @@ CV__DNN_INLINE_NS_BEGIN
|
||||
|
||||
extern bool DNN_DIAGNOSTICS_RUN;
|
||||
|
||||
#if HAVE_PROTOBUF
|
||||
#ifdef HAVE_PROTOBUF
|
||||
|
||||
using ::google::protobuf::RepeatedField;
|
||||
using ::google::protobuf::RepeatedPtrField;
|
||||
@ -3225,8 +3225,6 @@ void TFLayerHandler::handleFailed(const tensorflow::NodeDef& layer)
|
||||
|
||||
} // namespace
|
||||
|
||||
#endif //HAVE_PROTOBUF
|
||||
|
||||
Net readNetFromTensorflow(const String &model, const String &config)
|
||||
{
|
||||
return detail::readNetDiagnostic<TFImporter>(model.c_str(), config.c_str());
|
||||
@ -3276,5 +3274,27 @@ void writeTextGraph(const String& _model, const String& output)
|
||||
ofs.close();
|
||||
}
|
||||
|
||||
#else // HAVE_PROTOBUF
|
||||
|
||||
#define DNN_PROTOBUF_UNSUPPORTED() CV_Error(Error::StsError, "DNN/TF: Build OpenCV with Protobuf to import TensorFlow models")
|
||||
|
||||
Net readNetFromTensorflow(const String &, const String &) {
|
||||
DNN_PROTOBUF_UNSUPPORTED();
|
||||
}
|
||||
|
||||
Net readNetFromTensorflow(const char*, size_t, const char*, size_t) {
|
||||
DNN_PROTOBUF_UNSUPPORTED();
|
||||
}
|
||||
|
||||
Net readNetFromTensorflow(const std::vector<uchar>&, const std::vector<uchar>&) {
|
||||
DNN_PROTOBUF_UNSUPPORTED();
|
||||
}
|
||||
|
||||
void writeTextGraph(const String& _model, const String& output) {
|
||||
DNN_PROTOBUF_UNSUPPORTED();
|
||||
}
|
||||
|
||||
#endif // HAVE_PROTOBUF
|
||||
|
||||
CV__DNN_INLINE_NS_END
|
||||
}} // namespace
|
||||
|
Loading…
Reference in New Issue
Block a user