当前位置: 首页 > news >正文

一个小白的YOLOv10(MindYOLO)推理初尝试

前情提要

环境:笔记本、MindSpore、mindyolo、Jupyter Notebook
资源:mindyolo github、智能厨房 gitcode
小白工程师学习到了较繁琐的YOLOv10的新检测目标训练(非COCO数据集中已有目标),从github将mindyolo项目下载到了本地,但是看着里面复杂的文件夹结构,完全陌生四处关联的.yaml文件,散落各处的.py文件,找不到的dataset,无从下手。所以打算先从最简单的使用入手,争取实现图片简单推理,得到带分类和置信度的检测框。

实践过程

  • mindyolo github中下载完整压缩包,解压到jupyter notebook项目路径中,文件夹名为mindyolo
  • 根据教程,pip install mindyolo(其他必备包已安装)
  • 根据项目github的教程,下载了YOLOv10 N的yolov10n.yaml(Recipe)和yolov10n_500e_mAP383-c973023d.ckpt(Weights)文件,而实际上发现yolov10n文件可以在完整mindyolo(或mindyolo-master)文件夹中configs/yolov10中找到,且yolov10n.yaml中默认用到的coco.yaml和hyp.scratch.low.yaml文件也都在configs文件的正确相对位置,,所以单独下载的yolov10n.yaml后面并未被用到。
  • 按照教程提示通过predict.py来执行推理
    • 官网手册中命令:
      python demo/predict.py --config ./configs/yolov7/yolov7.yaml --weight=/path_to_ckpt/WEIGHT.ckpt --image_path /path_to_image/IMAGE.jpg
    • 自己修改调整后命令(Jupyter Notebook中运行所以增加了!开头):
      !python ./mindyolo/demo/predict.py --device_target "CPU" --config ./mindyolo/configs/yolov10/yolov10n.yaml --weight ./yolov10n_500e_mAP383-c973023d.ckpt --image_path ./human.jpg
  • 运行成功后,发现在Jupyter Notebook所在文件夹下出现了runs_infer文件夹,其中一路点击进去后即可看到加上了检测框的推理后图片
    human

问题与解决

  • 一开始按照教程,我只将predict.py路径、yaml路径、ckpt路径、jpg路径按现有情况修改了,但是报错说mindspore没有set_recursion_limit方法,定位到了predict.py中set_default_infer函数,将该句屏蔽了(tip: 发现gitcode的厨房推理中,这部分似乎也被删除;另外厨房示例中,原有的set_context方法比源文件多给了device_target和max_call_depth接口赋值。)
  • 后来运行提示非"Ascend"环境,所以找到了get_parser_infer函数部分,看到里面--device_target中默认是Ascend,所以想到将notebook中命令增加一个--device_target的参数,值为"CPU",最终运行成功

2025.11.14

http://www.rkmt.cn/news/49618.html

相关文章:

  • 文本生成器(AC自动机上DP)
  • ICLR2026 !SAM3重磅来袭:能“听懂人话”的分割模型,性能狂飙2倍
  • [题解]P11294 [NOISG 2022 Qualification] Tree Cutting
  • uniapp开发H5避坑指南
  • 精准把控VBAT,轻松规避电源设计99%陷阱
  • RAID(磁盘阵列)技术介绍
  • Sora2、Nano Banana AI批量生图视频实战,API低价调用和免费批量生成教程
  • 7 款最佳自托管 AI 工具,快速构建业务应用
  • 2025年果胶裂解酶批发厂家权威推荐榜单:果胶甲酯酶/果胶酶纤维素酶/果胶酶源头厂家精选
  • 鸿蒙应用开发实战:集成农历功能
  • 2025 最新波纹管厂家推荐榜:隧道 / 双壁 / 打孔 / 防结晶波纹管优质厂家权威甄选双壁波纹管/打孔波纹管/隧道打孔波纹管公司推荐
  • 2025年富氢水素机加盟生产厂家权威推荐榜单:家用富氢水机加盟/富氢水机加盟/智能富氢水机加盟源头厂家精选
  • 2025年木纹转印加工服务优质厂家推荐榜单TOP10
  • 2025年木纹转印加工服务全方位解析与优质厂家推荐榜单
  • Python操作多波段图像改变数据结构
  • 迈向人机共生文明:AI元人文构想的理论框架、架构与意义
  • dp problems
  • 2025年栏杆制作厂家综合实力排行榜:专业视角下的五大优选厂商
  • uniapp开发抖音小程序避坑指南
  • Windows安装MySQL,无服务模式,随用随有,一键初始化,可替换phpstudy_pro
  • 读书笔记:为什么数据库存储LOB数据时要选择「行内存储」?
  • LuatOS exgnss扩展库应用精讲:GNSS调试与优化实战全景!
  • 2025年北京雅思培训机构哪家强,高分提分 / 定制课程 / 封闭集训 / 小班教学 / 全程督学机构推荐哪家好
  • Keycloak会话标识符重用漏洞导致会话劫持风险分析
  • 2025 最新净化工程厂家推荐排行榜 实验室 / 手术室 / 医药 / 食品 / 厂房洁净工程优质服务商医药/食品净化工程/净化车间工程公司推荐
  • TurboWarp 部署云变量服务
  • JSC2023 Max Degree Sum
  • 2025年燃生物质有机热载体锅炉生产厂家权威推荐榜单:燃生物质热水锅炉/生物质专用锅炉/生物质热水锅炉源头厂家精选
  • 在线文档大全
  • AI大事记12:Transformer 架构——重塑 NLP 的革命性技能(下)