赞
踩
大家都知道,yolov8目前支持图像分类、目标检测、姿态估计、实例分割,而实例分割一般都是用命令行去预测图片,这个如果要引入pyqt5或着嵌入其他python模块有点麻烦,因此需要封装一个类去预测图像,这样我们可以轻松完成各种实例分割任务,封装好以后我们预测图片可以这么调用:
from Yolov8SegManager import *
ysm = Yolov8SegManager(weights=r'weights/yolov8s-seg.pt')
img = cv2.imread(r'images/bus.jpg')
results = ysm.inference_image(img)
result_img = ysm.draw_image2(results)
cv2.imshow('result', result_img)
cv2.waitKey(0)
可见我们只需要少量代码即可完成实例分割任务。关于yolov8实例分割细节视频介绍参考:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。