opencv

Open Source Computer Vision Library https://opencv.org/

Yuantao Feng bc0618b688 Merge pull request #25582 from fengyuentau:dnn/dump_pbtxt Current net exporter `dump` and `dumpToFile` exports the network structure (and its params) to a .dot file which works with `graphviz`. This is hard to use and not friendly to new user. What's worse, the produced picture is not looking pretty. dnn: better net exporter that works with netron #25582 This PR introduces new exporter `dumpToPbtxt` and uses this new exporter by default with environment variable `OPENCV_DNN_NETWORK_DUMP`. It mimics the string output of a onnx model but modified with dnn-specific changes, see below for an example. ![image](https://github.com/opencv/opencv/assets/17219438/0644bed1-da71-4019-8466-88390698e4df) ## Usage Call `cv::dnn::Net::dumpToPbtxt`: ```cpp TEST(DumpNet, dumpToPbtxt) { std::string path = "/path/to/model.onnx"; auto net = readNet(path); Mat input(std::vector<int>{1, 3, 640, 480}, CV_32F); net.setInput(input); net.dumpToPbtxt("yunet.pbtxt"); } ``` Set `export OPENCV_DNN_NETWORK_DUMP=1` ```cpp TEST(DumpNet, env) { std::string path = "/path/to/model.onnx"; auto net = readNet(path); Mat input(std::vector<int>{1, 3, 640, 480}, CV_32F); net.setInput(input); net.forward(); } ``` --- Note: - `pbtxt` is registered as one of the ONNX model suffix in netron. So you can see `module: ai.onnx` and such in the model. - We can get the string output of an ONNX model with the following script ```python import onnx net = onnx.load("/path/to/model.onnx") net_str = str(net) file = open("/path/to/model.pbtxt", "w") file.write(net_str) file.close() ``` ### 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 - [ ] There is a reference to the original bug report and related work - [ ] 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		8 months ago
.github	Added Ubuntu 24.04 to regular CI.	8 months ago
3rdparty	Merge pull request #25443 from asmorkalov:as/kleidicv_hal	8 months ago
apps	Merge pull request #25582 from fengyuentau:dnn/dump_pbtxt	8 months ago
cmake	Fix zlib-ng version parse	9 months ago
data	…
doc	Merge pull request #25443 from asmorkalov:as/kleidicv_hal	8 months ago
include	…
modules	Merge pull request #25582 from fengyuentau:dnn/dump_pbtxt	8 months ago
platforms	Added FaceDetectorYN to OpenCV JS API.	9 months ago
samples	Update text_detection.cpp	9 months ago
.editorconfig	…
.gitattributes	…
.gitignore	…
CMakeLists.txt	Merge pull request #25585 from FantasqueX:add-policy-0148	8 months ago
CONTRIBUTING.md	…
COPYRIGHT	…
LICENSE	…
README.md	Space mistake in README.md	11 months ago
SECURITY.md	…

README.md

OpenCV: Open Source Computer Vision Library

Resources

Homepage: https://opencv.org
- Courses: https://opencv.org/courses
Docs: https://docs.opencv.org/4.x/
Q&A forum: https://forum.opencv.org
- previous forum (read only): http://answers.opencv.org
Issue tracking: https://github.com/opencv/opencv/issues
Additional OpenCV functionality: https://github.com/opencv/opencv_contrib
Donate to OpenCV: https://opencv.org/support/

Contributing

Please read the contribution guidelines before starting work on a pull request.

Summary of the guidelines:

One pull request per issue;
Choose the right base branch;
Include tests and documentation;
Clean up "oops" commits before submitting;
Follow the coding style guide.

Additional Resources

Submit your OpenCV-based project for inclusion in Community Friday on opencv.org
Subscribe to the OpenCV YouTube Channel featuring OpenCV Live, an hour-long streaming show
Follow OpenCV on LinkedIn for daily posts showing the state-of-the-art in computer vision & AI
Apply to be an OpenCV Volunteer to help organize events and online campaigns as well as amplify them
Follow OpenCV on Mastodon in the Fediverse
Follow OpenCV on Twitter
OpenCV.ai: Computer Vision and AI development services from the OpenCV team.