18 lines
533 B
Python
18 lines
533 B
Python
|
from ultralytics import YOLO
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
# 載入模型
|
||
|
|
model = YOLO('yolov8s.yaml').load('yolov8s.pt') # 從 YAML 配置並加載權重
|
||
|
|
|
||
|
|
# 訓練參數設定
|
||
|
|
img_size = 640 # 訓練影像大小
|
||
|
|
num_epochs = 300 # 訓練次數
|
||
|
|
|
||
|
|
# 訓練模型
|
||
|
|
print("訓練開始")
|
||
|
|
model.train(
|
||
|
|
data='E:/AP/YOLODataset/dataset.yaml', # 資料集的配置文件
|
||
|
|
imgsz=img_size, # 設置影像大小
|
||
|
|
epochs=num_epochs # 訓練的 epoch 次數
|
||
|
|
)
|
||
|
|
print("訓練結束")
|