import torch
from torch.serialization import add_safe_globals
from models.yolo import Model

add_safe_globals([Model])  # 允許 PyTorch 安全地載入 YOLOv5 模型類別

model = torch.hub.load('ultralytics/yolov5', 'custom', path = 'runs/train/exp/weights/best.pt')		# 載入自行訓練的 YOLOv5 模型
img = 'datasets/road/images/train/road6.png'		# 影像來源

model.iou = 0.5             # 設定 IoU 門檻值
model.conf = 0.5            # 設定信心門檻值

results = model(img)		# 進行物件偵測

results.print()             # 顯示結果摘要
results.show()				# 顯示結果圖片
results.save()				# 儲存結果圖片

print(results.pandas().xyxy[0])    # 顯示物件列表