import torch
from torch.serialization import add_safe_globals
from models.yolo import Model		# 這行要在 yolov5 專案目錄中執行

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

model = torch.hub.load('ultralytics/yolov5', 'yolov5n')	# 從 PyTorch Hub 下載 YOLOv5s 預訓練模型，可選用的模型有 yolov5s, yolov5m, yolov5x 等
img = 'https://ultralytics.com/images/zidane.jpg'		# 影像來源，支援檔案、路徑、PIL、OpenCV, NumPy, list 等
results = model(img)		# 進行物件偵測
results.print()				# 顯示結果摘要

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

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