Rust 语言从入门到实战
唐刚
Rust 语言中文社区联合创始人
5266 人已学习
新⼈⾸单¥59
登录后,你可以任选4讲全文学习
课程目录
已完结/共 36 讲
Rust 语言从入门到实战
15
15
1.0x
00:00/00:00
登录|注册

24|Rust图像识别:利用YOLOv8识别对象

你好,我是 Mike。这节课我们来学习如何使用 Rust 对图片中的对象进行识别。
图像识别是计算机视觉领域中的重要课题,而计算机视觉又是 AI 的重要组成部分,相当于 AI 的眼睛。目前图像识别领域使用最广泛的框架是 YOLO,现在已经迭代到了 v8 版本。而基于 Rust 机器学习框架 Candle,可以方便地实现 YOLOv8 算法。因此,这节课我们继续使用 Candle 框架来实现图片的识别。
Candle 框架有一个示例,演示了 YOLOv8 的一个简化实现。我在此基础上,将这个源码中的示例独立出来,做成了一个单独的项目,方便你学习(查看代码地址)。
注:这节课的代码适用于 candle_core v0.3 版本。

YOLO 简介

YOLO(You Only Look Once)是一种目标检测算法,它可以在一次前向传递中检测出图像中的所有物体的位置和类别。因为只需要看一次,YOLO 被称为 Region-free 方法,相比于 Region-based 方法,YOLO 不需要提前找到可能存在目标的区域(Region)。YOLO 在 2016 年被提出,发表在计算机视觉顶会 CVPR(Computer Vision and Pattern Recognition)上。YOLO 对整个图片进行预测,并且它会一次性输出所有检测到的目标信息,包括类别和位置。
确认放弃笔记?
放弃后所记笔记将不保留。
新功能上线,你的历史笔记已初始化为私密笔记,是否一键批量公开?
批量公开的笔记不会为你同步至部落
公开
同步至部落
取消
完成
0/2000
荧光笔
直线
曲线
笔记
复制
AI
  • 深入了解
  • 翻译
    • 英语
    • 中文简体
    • 中文繁体
    • 法语
    • 德语
    • 日语
    • 韩语
    • 俄语
    • 西班牙语
    • 阿拉伯语
  • 解释
  • 总结

本文介绍了如何使用Rust和YOLOv8算法进行图像识别。YOLOv8是一种高效的目标检测算法,能够在一次前向传递中检测出图像中的所有物体的位置和类别。文章详细介绍了YOLOv8的模型结构和能力,并提供了使用Candle框架实现图片识别的示例。读者可以通过指定参数选择不同的模型,实现物体探测和姿势探测。文章还提供了具体的代码示例和运行效果展示,帮助读者快速了解如何使用Rust和YOLOv8进行图像识别。此外,文章还介绍了如何选择不同的预训练模型和图片来测试识别效果,并鼓励读者分享实验结果。总的来说,本文为读者提供了全面的指导,使其能够快速上手利用Rust和YOLOv8进行图像识别。同时,文章还强调了Rust在AI领域的应用前景,为读者展示了Rust在图像识别领域的潜力和优势。

仅可试看部分内容,如需阅读全部内容,请付费购买文章所属专栏
《Rust 语言从入门到实战》
新⼈⾸单¥59
立即购买
登录 后留言

全部留言(10)

  • 最新
  • 精选
  • Geek_118351
    老师你好,会考虑出一个针对视频流的目标识别课程吗。

    作者回复: 看极客时间这边的安排哈。 对于rust ai的解决方案,我后续会持续输出,频率不定。

    2024-03-11归属地:四川
  • 渡鸦10086
    网页下载模型到本地后通过--models参数即可使用本地模型,比如 cargo run --release -- assets/football.jpg --which m --model ./model/yolov8m.safetensors

    作者回复: 👍

    2024-02-01归属地:陕西
  • Geek_e72251
    老师可以贴一段怎么从本地加载 yolo 模型的代码吗?实在下载不下来😮‍💨

    作者回复: 你可以在浏览器里面打开这个 https://huggingface.co/lmz/candle-yolo-v8/tree/main 下载模型文件,然后改 https://github.com/rustai-solutions/slint-yolov8-demo/blob/main/src/yolov8engine/mod.rs#L307C41-L307C41 实例的 model 字段为具体的路径,就是你本地目录的Path。正好可以练手一下哈。

    2024-01-11归属地:广东
    2
  • Geek_e72251
    Error: request error: https://huggingface.co/lmz/candle-yolo-v8/resolve/main/yolov8m.safetensors: Connection Failed: Connect error: connection timed out Caused by: 0: https://huggingface.co/lmz/candle-yolo-v8/resolve/main/yolov8m.safetensors: Connection Failed: Connect error: connection timed out 1: connection timed out 一直下不来这个文件,可以提前下载下来然后放到项目里面吗?浏览器可以正常下载

    作者回复: 可以像前面一节一样,用hf的工具从镜像下载。或者开全局代理下载

    2024-01-05归属地:广东
    5
  • 凤 梨 🍍
    pytorch怎么转safetensors,没工具下载不了外面的模型

    作者回复: 我代码里面有, 用 huggingface-cli 下载,指定镜像. 可以下载的. 转换格式的话可以用这个方案: https://github.com/ToluClassics/candle-tutorial

    2023-12-30归属地:广东
    2
  • unistart
    老师,我有一个问题就是猫猫那张图执行姿势探测任务时无法正确识别,这是为什么啊? cargo run --release -- assets/cats.jpg --model candle-yolo-v8/yolov8x-pose.safetensors --which x --task pose Compiling candle_demo_yolov8 v0.1.0 (E:\Project\rust-jikeshijian\24-candle_yolov8) Finished release [optimized] target(s) in 12.69s Running `target\release\candle_demo_yolov8.exe assets/cats.jpg --model candle-yolo-v8/yolov8x-pose.safetensors --which x --task pose` Running on CPU, to run on GPU, build this example with `--features cuda` model loaded processing "assets/cats.jpg" generated predictions Tensor[dims 56, 6300; f32] writing "assets/cats.pp.jpg"

    作者回复: 因为姿势探测好像只支持人。

    2023-12-25归属地:湖南
  • 蕨火
    同问,不联网怎么做?

    作者回复: 训练yolo跟联网关系不大,可以先把预训练模型下载下来,准备好环境,再把网络断掉训练。训练的过程本身不依赖于网络,只是前期资料(初始模型文件,你的数据集,rust app编译好,等等)准备你得准备好。

    2023-12-20归属地:青海
  • My dream
    用rust怎么训练录像资源啊?

    作者回复: 这块儿还没有示例,简单的说,pytorch是怎么训练的,在candle中就对应的实现就行了。原理都是神经网络。Rust在AI这块儿还有大量的工作要做,而Rust社区后面会在这块儿投入巨大的精力。欢迎持续关注。

    2023-12-19归属地:四川
  • My dream
    如果我们的电脑不联网的情况下,用yolo训练图片资源啊?

    作者回复: 训练yolo跟联网关系不大,可以先把预训练模型下载下来,准备好环境,再把网络断掉训练。训练的过程本身不依赖于网络,只是前期资料准备你得准备好。

    2023-12-19归属地:四川
    2
  • My dream
    怎么使用yolo训练图片?请老师请一下

    作者回复: 你指的在自己的图片训练集上进行训练吗?目前网上有很多相关资料:https://zhuanlan.zhihu.com/p/650002512 https://blog.csdn.net/weixin_45921929/article/details/132450479 https://blog.csdn.net/qq_40716944/article/details/128648001 Candle这边资料现在非常少,毕竟是最新刚出的东西,还在飞速发展中。不过,后面我会在这块儿持续输出相关内容,包含yolov8的训练。欢迎持续关注。

    2023-12-19归属地:四川
收起评论
显示
设置
留言
10
收藏
沉浸
阅读
分享
手机端
快捷键
回顶部