当前位置:   article > 正文

yolov8 如何将分割的mask掩模,显示读取出来 ,Masks类型 转换为 ndarray_yolo掩膜

yolo掩膜

将分割的mask掩模,显示读取


#识别图片
model = YOLO('yolov8n-seg');
results = model("your_image_path")
#关键在于将Masks 转换为 ndarray
mask_raw = results[0].masks[0].cpu().data.numpy().transpose(1,2,0) 
cv2.imshow("result",mask_raw);

#=======================分割线=========================================
#识别视频流
sourse = "./list.streams" #list.streams文件,用来存放流媒体地址,如rstp://localhost/test
results = model(sourse,stream=True)
for result in results:
	if result.mask is not None:
		mask_raw = result.masks[0].cpu().data.numpy().transpose(1,2,0)
		for mask in result.masks:
			mask_raw += mask.cpu().data.numpy().transpose(1,2,0)
		cv2.imshow("result",mask_raw)


  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20

效果如下:
在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/275869
推荐阅读
相关标签
  

闽ICP备14008679号