当前位置:   article > 正文

[C++]yolov8的onnx模型加密方法保护自己模型和版权

[C++]yolov8的onnx模型加密方法保护自己模型和版权

【官方框架地址】

https://github.com/ultralytics/ultralytics
【算法介绍】

Yolov8 是一种先进的对象检测模型,基于深度学习技术,广泛应用于计算机视觉任务。它使用 ONNX(Open Neural Network Exchange)格式,使得模型在不同的深度学习框架之间具有更好的互操作性。然而,直接使用 ONNX 格式的模型存在一定的安全隐患,因此对模型进行加密就显得尤为重要。

对 Yolov8 的 ONNX 模型进行加密,主要是为了保护模型的机密性和完整性,防止未经授权的访问和使用。常见的加密方法包括使用密码加密、哈希函数、数字签名等技术。这些方法可以在模型传输、存储和使用过程中提供一定的安全保障。

具体来说,当模型需要从一个环境传输到另一个环境时,可以使用密码加密的方式对模型进行加密,确保模型在传输过程中的安全。在模型存储时,可以使用哈希函数和数字签名等技术,验证模型的完整性和来源,防止模型被篡改或替换。在使用模型时,可以通过身份验证和访问控制等机制,确保只有经过授权的人员才能访问和使用模型。

需要注意的是,对 Yolov8 的 ONNX 模型进行加密可能会对模型的性能产生一定的影响。因此,在选择加密方法时,需要根据实际需求和场景进行权衡,尽可能地平衡安全性和性能的需求。

总之,对 Yolov8 的 ONNX 模型进行加密是一个重要的安全措施,可以有效地保护模型的机密性和完整性,防止未经授权的访问和使用。在选择加密方法时,需要根据实际需求和场景进行权衡,尽可能地平衡安全性和性能的需求。

对于onnx加密我们采用对称加密方式,采用二进制加密除非有密钥否则很难被破解,这样可以使我们的花费大量人力和物力以及成本得到安全保障。
【效果展示】

加密后缀为yolov8.firc您可以改成任何后缀不影响模型


【实现部分代码】

  1. #include <iostream>
  2. #include <opencv2/opencv.hpp>
  3. #include <math.h>
  4. #include "yolov8_onnx.h"
  5. #include "OnnxEncry.h"
  6. #include <time.h>
  7. using namespace std;
  8. using namespace cv;
  9. using namespace dnn;
  10. int main()
  11. {
  12. string img_path = "C:\\Users\\Administrator\\Desktop\\yolov8-onnxruntime-encry\\x64\\Debug\\zidane.jpg";
  13. string model_path = "C:\\Users\\Administrator\\Desktop\\yolov8-onnxruntime-encry\\x64\\Debug\\\\yolov8s.onnx";
  14. string encode_path = "C:\\Users\\Administrator\\Desktop\\yolov8-onnxruntime-encry\\x64\\Debug\\yolov8s.firc";
  15. /* OnnxEncry oe;
  16. oe.EncryOnnx(model_path,encode_path);
  17. return 0;*/
  18. Mat img = imread(img_path);
  19. Yolov8Onnx det;
  20. det.ReadModelEncode(encode_path, false);
  21. vector<Scalar> color;
  22. srand(time(0));
  23. for (int i = 0; i < 80; i++)
  24. {
  25. int b = rand() % 256;
  26. int g = rand() % 256;
  27. int r = rand() % 256;
  28. color.push_back(Scalar(b, g, r));
  29. }
  30. vector<OutputSeg> result;
  31. if (det.OnnxDetect(img, result))
  32. {
  33. DrawPred(img, result, det._className, color);
  34. }
  35. else
  36. {
  37. cout << "Detect nothing!" << endl;
  38. }
  39. return 0;
  40. }


【视频演示】

https://www.bilibili.com/video/BV1jw41177mQ/
【源码下载】

https://download.csdn.net/download/FL1623863129/88784066
【测试环境】

 vs2019,onnxruntime==1.12.0,opencv==4.7.0

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/秋刀鱼在做梦/article/detail/942417
推荐阅读
相关标签
  

闽ICP备14008679号