360_layout_analysis.py 221 B

12345678
  1. from ultralytics import YOLO
  2. image_path = '' # 待预测图片路径
  3. model_path = '' # 权重路径
  4. model = YOLO(model_path)
  5. result = model(image_path, save=True, conf=0.5, save_crop=False, line_width=2)
  6. print(result)