码农知识堂 - 1000bd
  •   Python
  •   PHP
  •   JS/TS
  •   JAVA
  •   C/C++
  •   C#
  •   GO
  •   Kotlin
  •   Swift
  • c#调用CUDA执行YOLOV5对象检测


    c#使用调YOLOV5对象检测,并调用CUDA进行计算

    1.CUDA版本11.2

    2.cuDNN用cudnn-windows-x86_64-8.9.3.28_cuda11-archive

    记得把压缩包的三个文件夹放到cuda根目录下覆盖

    3.Microsoft.ML.OnnxRuntime.Gpu要使用1.13.1,如果版本太新,SessionOptions会报错。

    1. private SessionOptions GetSessionOptions()
    2. {
    3. bool supportGPU = (Environment.GetEnvironmentVariable("CPAI_MODULE_SUPPORT_GPU") ?? "true").ToLower() == "true";
    4. SessionOptions sessionOpts = new SessionOptions();
    5. string[]? providers = null;
    6. try
    7. {
    8. providers = OrtEnv.Instance().GetAvailableProviders();
    9. }
    10. catch
    11. {
    12. }
    13. sessionOpts.AppendExecutionProvider_CUDA();//显卡用这个
    14. //sessionOpts.AppendExecutionProvider_CPU();
    15. return sessionOpts;
    16. }
    17. //初始化onnx模型的代码
    18. SessionOptions sessionOpts = GetSessionOptions();
    19. _scorer = new YoloScorer("Assets/Weights/yolov5n.onnx", sessionOpts);
    20. _capture = new VideoCapture("http://192.168.151.130:8080/video"); // 手机摄像头流媒体

    但是最终在2060笔记本显卡上发现 CUDA速度不如AMD的CPU快。。。

  • 相关阅读:
    Vue路由守卫有哪些,怎么设置,有哪些使用场景?
    Pytorch中的广播机制
    LegalAI领域大规模预训练语言模型的整理、总结及介绍(持续更新ing…)
    组件安全概述
    RISC-V知识总结 —— 指令集
    DeepStream--测试lpdnet车牌检测模型
    Java 面试题:Java 的文件拷贝方式有几种?哪一种最高效?
    QFileDialog 文件对话框
    maven安装及配置(详细版)
    RabbitMQ之交换机
  • 原文地址:https://blog.csdn.net/ccagy/article/details/133847381
  • 最新文章
  • 攻防演习之三天拿下官网站群
    数据安全治理学习——前期安全规划和安全管理体系建设
    企业安全 | 企业内一次钓鱼演练准备过程
    内网渗透测试 | Kerberos协议及其部分攻击手法
    0day的产生 | 不懂代码的"代码审计"
    安装scrcpy-client模块av模块异常,环境问题解决方案
    leetcode hot100【LeetCode 279. 完全平方数】java实现
    OpenWrt下安装Mosquitto
    AnatoMask论文汇总
    【AI日记】24.11.01 LangChain、openai api和github copilot
  • 热门文章
  • 十款代码表白小特效 一个比一个浪漫 赶紧收藏起来吧!!!
    奉劝各位学弟学妹们,该打造你的技术影响力了!
    五年了,我在 CSDN 的两个一百万。
    Java俄罗斯方块,老程序员花了一个周末,连接中学年代!
    面试官都震惊,你这网络基础可以啊!
    你真的会用百度吗?我不信 — 那些不为人知的搜索引擎语法
    心情不好的时候,用 Python 画棵樱花树送给自己吧
    通宵一晚做出来的一款类似CS的第一人称射击游戏Demo!原来做游戏也不是很难,连憨憨学妹都学会了!
    13 万字 C 语言从入门到精通保姆级教程2021 年版
    10行代码集2000张美女图,Python爬虫120例,再上征途
Copyright © 2022 侵权请联系2656653265@qq.com    京ICP备2022015340号-1
正则表达式工具 cron表达式工具 密码生成工具

京公网安备 11010502049817号