Yolo-World初步使用

Yolo v8目前已经支持Yolo-World,整理一下初步使用步骤。

使用步骤

1 先下载Yolo-World的pt文件,下载地址:GitHub - AILab-CVC/YOLO-World: [CVPR 2024] Real-Time Open-Vocabulary Object Detection

官网应该是点这里(有个笑脸)下载,我一直刷不出来界面,所以不确定:

我自己用的是在其他网友提供的地址下载的,应该是V1版本(yolov8-world.pt),目前已经出了V2版本。

2 新建一个python脚本,叫testYolo-World.py,这里我们使用Yolo v8自带的bus.jpg进行测试。

python 复制代码
from ultralytics import YOLOWorld


# 目标检测
# Load a model
model = YOLOWorld('data/yolov8s-world.pt')  # load a pretrained model

# Define custom classes
model.set_classes(['glasses'])

#model.save("data/yolov8s-world2.pt")

# Run inference on 'bus.jpg' with arguments
results = model.predict('data/bus.jpg',save=True, imgsz=640)

运行过程中,会提示缺失CLIP库,然后等待一会,会自动安装。安装成功后,再次运行一遍,会出结果,能检测到glasses.

但是尝试了其他类别,'tree' 'window' 'clothes' 'shoes',除了'shoes',其他都检测不出来,不知道V2效果会不会好一点。

如果注释掉设置类别这句话,能检测到的类别和Yolo v8自带的pt文件检测类别基本一致,置信度很高,但是有一个错误,多检测出来一个person。

相关推荐
XIAO·宝1 天前
深度学习------YOLOV3
人工智能·深度学习·yolo
newxtc1 天前
【广州公共资源交易-注册安全分析报告-无验证方式导致安全隐患】
开发语言·selenium·安全·yolo
总有刁民想爱朕ha1 天前
YOLO目标检测:一种用于无人机的新型轻量级目标检测网络
yolo·目标检测·无人机
baole9631 天前
YOLOv4简单基础学习
学习·yolo·目标跟踪
我叫侯小科1 天前
YOLOv4:目标检测界的 “集大成者”
人工智能·yolo·目标检测
麒羽7601 天前
YOLOv4:目标检测领域的 “速度与精度平衡大师”
yolo·目标检测·目标跟踪
前网易架构师-高司机1 天前
鸡蛋质量识别数据集,可识别染血的鸡蛋,棕色鸡蛋,钙沉积鸡蛋,污垢染色的鸡蛋,白鸡蛋,平均正确识别率可达89%,支持yolo, json, xml格式的标注
yolo·分类·数据集·缺陷·鸡蛋
Python图像识别1 天前
73_基于深度学习的水面漂浮垃圾检测系统(yolo11、yolov8、yolov5+UI界面+Python项目源码+模型+标注好的数据集)
python·深度学习·yolo
JANGHIGH2 天前
YOLO系列——Ubuntu20.04下通过conda虚拟环境源码安装opencv-4.10与opencv_contrib-4.10.0
opencv·yolo·conda
放羊郎2 天前
面向不同障碍物的自主导航策略
yolo·slam·识别·平滑·语义地图