YOLO-v8-seg实例分割使用

news/2024/4/27 1:46:54

最近需要实例分割完成一些任务,一直用的SAM(segment anything)速度慢,找一个轻量分割模型。

1. YOLO-v8-seg使用

git clone https://github.com/ultralytics/ultralytics.git

cd ultralytics

vim run.py

from ultralytics import YOLO# Load a model
model = YOLO('yolov8l-seg.pt')  # load an official model# Predict with the model
results = model('test.jpg')  # predict on an image

就这么几行,这就行了?代码量也太少了。

2 效果

效果也还不错

速度:

3. 对结果results的解析

可以看官方详细介绍:预测 -Ultralytics YOLOv8 文档

img = cv2.imread(img_path)
imgray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
print(img.shape)
for result in results:boxes = result.boxes  # Boxes object for bounding box outputsmasks = result.masks  # Masks object for segmentation masks outputsfor mask in masks:print(mask.xy)print(type(mask.xy))points = np.array(mask.xy, dtype=np.int32)cv2.polylines(img, [points], isClosed=True, color=(0, 255, 0), thickness=2)keypoints = result.keypoints  # Keypoints object for pose outputsprobs = result.probs  # Probs object for classification outputs#result.show()  # display to screenresult.save(filename='result.jpg')  # save to disk
cv2.imwrite('test.jpg',img)

很好用,希望对你有帮助!

THE END!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.cpky.cn/p/11101.html

如若内容造成侵权/违法违规/事实不符,请联系我们进行投诉反馈,一经查实,立即删除!

相关文章

网络工程师笔记15(OSPF协议-2)

OSPF协议 OSPF是典型的链路状态路由协议,是目前业内使用非常广泛的 IGP 协议之一。 Router-ID(Router ldentifier,路由器标识符),用于在一个 OSPF 域中唯一地标识一台路由器。Router-ID 的设定可以通过手工配置的方式,或使用系统自…

【Linux】Linux基本开发工具(yum) (vi/vim)的使用

本文章内容: 学习yum工具,进行软件安装掌握vim编辑器使用 Linux 软件包管理器 yum 什么是软件包 在Linux下安装软件, 一个通常的办法是下载到程序的源代码, 并进行编译, 得到可执行程序.但是这样太麻烦了, 于是有些人把一些常用的软件提前编译好, 做成…

酷炫的粒子动态表白HTML源码

源码介绍 酷炫的粒子动态表白HTML源码,自己自定义文字,动态组合文字,进行表白,喜欢的朋友可以下载使用,很不错的表白HTML代码 下载地址 酷炫的粒子动态表白HTML源码

css3鼠标悬停图片特效,图片悬停效果源码

特效介绍 css3鼠标悬停图片特效,图片悬停效果源码,可以在网页上面作为自己的动态加载名片,放到侧边栏或者网站合适的位置即可 动态效果 代码下载 css3鼠标悬停图片特效,图片悬停效果源码

Python|OpenCV-基于OpenCV实现给图像增加噪点(11)

前言 本文是该专栏的第12篇,后面将持续分享OpenCV计算机视觉的干货知识,记得关注。 在图像处理中,给图像添加噪点是一个常见的预处理步骤,用于模拟真实世界中的噪声条件,从而评估或测试图像处理算法的性能。OpenCV是一个强大的计算机视觉库,它提供了许多函数和方法来处理…

外包干了5年,技术退步明显.......

先说一下自己的情况,大专生,18年通过校招进入杭州某软件公司,干了接近4年的功能测试,今年年初,感觉自己不能够在这样下去了,长时间呆在一个舒适的环境会让一个人堕落! 而我已经在一个企业干了四年的功能测…