最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

python - "How to hide labels and probability in YOLOv8 while saving predictions?" - Stack Overflow

programmeradmin1浏览0评论

"How to hide labels and probability in YOLOv8 while saving predictions?"

from ultralytics import YOLO
from pathlib import Path

model = YOLO("yolov8m-seg.pt")

results = model.train(
data="D:/a)tutot_seg/YOLO_dataset/data.yaml",
epochs=3,
imgsz=1280,
batch=1
)

image_folder = Path("D:/A)seg23/COCO_to_YOLOv8/YOLO_dataset/validation/images")
image_paths = list(image_folder.glob("\*.jpg"))

results = model.predict(source=image_paths, verbose=False)

output_folder = Path("D:/A)seg23/COCO_to_YOLOv8/YOLO_dataset/predictions")
output_folder.mkdir(parents=True, exist_ok=True)

for i, result in enumerate(results):
annotated_image = result.plot(labels=False, conf=False)  # Убираем метки и вероятность
output_path = output_folder / f"pred\_{i}.jpg"
result.save(output_path)  # Сохраняем обработанное изображение

print("Предсказания сохранены в:", output_folder)

help with the label and sant remove this

发布评论

评论列表(0)

  1. 暂无评论