如何使用YOLO-World做目标检测

科技   2024-11-18 08:46   重庆  

点击下方卡片,关注“机器视觉与AI深度学习

视觉/图像重干货,第一时间送达!

介绍

    YOLO-World 是一种尖端的目标检测系统,在识别图像中物体的方式上开辟了新天地。与需要预定义类别的传统检测器不同,YOLO-World可以识别各种物体,甚至是以前从未见过的物体——这是“开放词汇”检测!这是通过对海量数据集的预训练和对语言的巧妙运用来实现的。可以把它想象成用文字向 YOLO-World 展示什么是“停车标志”,然后它就可以实时找到停车标志,而无需任何进一步的训练。这使得它速度超快并且适用于许多应用程序。

如何使用YOLO-World

    第1步:安装必要的库
pip install ultralytics

    第2步:导入库

from ultralytics import YOLOWorld

    第3步:选择模型

model = YOLOWorld("yolov8x-worldv2.pt")
https://docs.ultralytics.com/de/models/yolo-world/#available-models-supported-tasks-and-operating-modes

    在上面链接中,您可以比较不同的型号并权衡各自的优缺点。在本例中我们选择了yolov8x-worldv2.pt

    第4步:使用YOLO-World检测图像中的目标

model.set_classes(["any", "class", "you", "want"])results = model.predict('YourImagePath')

    如果你想检测特定的类,你可以在 model.set_classes 函数列表中编写任何类

    第5步:保存并绘制结果

# Show resultsresults[0].show()
参考链接:
YOLO-World论文: 
https://arxiv.org/pdf/2401.17270v2.pdf
YOLO-World Github: 
https: //github.com/AILab-CVC/YOLO-World

—THE END—

觉得有用,麻烦给个赞和在看 

机器视觉与AI深度学习
专注于机器视觉、AI、深度学习等技术最新资讯、实战内容及应用案例的分享,交流!
 最新文章