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)