本站资源来自互联网用户收集发布,仅供用于学习和交流。如有侵权之处,请联系站长并出示版权证明以便删除,敬请谅解!(邮箱:yunmengyaa@gmail.com)

查看: 80|回复: 0

[电脑应用] Python的ultralytics YOLO V8/11标注 训练 模型转换测试工具源码

[复制链接]
  • TA的每日心情
    难过
    17 小时前
  • 签到天数: 908 天

    [LV.10]以坛为家III

    7603

    主题

    518

    帖子

    4万

    积分

    管理员

    盛年不重来,一日难再晨。

    UID
    1
    云币
    7187
    金钱
    21898
    好评
    567
    性别
    在线时间
    2155 小时
    注册时间
    2019-8-21

    管理员勋章最佳新人活跃会员热心会员推广达人宣传达人灌水之王突出贡献优秀版主荣誉管理论坛元老财富大使热爱篮球国庆勋章灌水大神帖子大侠以坛为家劳动节勋章虎年纪念勋章夜猫子搬运大神音乐无限游戏大全

    发表于 前天 23:50 | 显示全部楼层 |阅读模式 IP:福建泉州


    含完整的标注/训练/调用测试系统,继上个版本进行了大量优化,这个版本确定可以完全打包并且运行,需要安装的库文件全部已经注明:
    1 添加了自动标注(载入模型自动标注信息)详细的快捷按键
    2 使用了ultralytics api方式训练,并且支持了windows的多进程(此处感谢给予帮助的朋友)
    3 移除了动态批大小,这个参数在转成onnx时调用需要特殊处理不然奔溃所以移除了
    4 模型转换时添加了矩形限定宽高参数,防止正方形参数导致检测宽度过宽,另外兼容其他E模块的调用,这里需要注意如果使用自己写的代码调用转换出来的onnx模型,需要修改代码支持修改的矩形宽高
    5 添加了指定模型继续训练
    6 训练参数添加了更多的参数如数据增强,矩形,时间限定等
    。。。。。。。
    所有窗口可以同时操作,测试速度还是很快的onnx单线调用4-5ms左右,貌似ncnn对多线程的调用速度更快
    需要注意ultralytics 8.3.0以前的版本支持yolov8 以后的版本支持yolov11和v8,只需要切换版本和模型(应该是)
    其他还有些细节优化忘掉了,有些细节地方有点小问题如目录选择,不过没有影响自己手动选中下就行,基本满足正常使用。有问题自己解决。
    记得安装vc++组件 显卡驱动,运行不了再装cuda(应该可能也许理论上是不用安装的),总之自行解决,源码百分百确定可以运行没有报错!

    CUDA 版本
    最低驱动版本

    CUDA 12.4
    550.54.14

    CUDA 12.3
    545.23.08

    CUDA 12.2
    537.13

    CUDA 12.1
    530.30.02

    CUDA 12.0
    525.60.13

    驱动版本检查:
    Windows:在命令行运行 nvidia-smi,查看右上角的 Driver Version
    操作系统要求:
    Windows:需 Windows 10/11(64 位)。
    打包出来的软件第一次启动速度慢,机器配置差启动时间可能会长点!
    没有使用视频自己研究,不接单!
    为什么开源,因为热爱。
    地址:
    游客,如果您要查看本帖隐藏内容请回复

    楼主热帖
    回复

    使用道具 举报

    您需要登录后才可以回帖 登录 | 注册账号

    本版积分规则

    手机版|小黑屋|秋月社区 |网站地图

    GMT+8, 2025-6-4 17:46 , Processed in 0.048728 second(s), 28 queries , Gzip On.

    Powered by Qiu Yue X3.4 Licensed

    © 2019-2024 Comsenz Inc.