yolo11-seg 推理测试infer

import cv2

from ultralytics import YOLO

img = "image_00209.jpg"

img = cv2.imread(img)

加载模型

model = YOLO("last.pt")

results = model.predict(source=img, conf=0.6, show=False)

for i in results:

fam = i.plot() #提取推理后的图像

cv2.imshow("outcome", fam) #显示结果

cv2.waitKey(0)

或者读取文件夹：

img = '/data/yao_bag/img1/'

img = cv2.imread(img)

加载模型

model = YOLO("last.pt")

results = model.predict(source=img, conf=0.6,show_boxes=False)

for i in results:

fam = i.plot(boxes=False) #提取推理后的图像

masks = i.masks

cv2.imshow("outcome", fam) #显示结果

cv2.waitKey(0)

或者：

import cv2,glob

from ultralytics import YOLO

img = '/img1/'

for img in glob.glob('{}/*jpg'.format(img)):

img = cv2.imread(img)

加载模型

model = YOLO("last.pt")

results = model.predict(source=img, conf=0.6,show_boxes=False)

for i in results:

fam = i.plot(boxes=False) #提取推理后的图像

masks = i.masks

cv2.imshow("outcome", fam) #显示结果

cv2.waitKey(0)