21 lines
1.0 KiB
Python
21 lines
1.0 KiB
Python
|
from ultralytics import YOLO
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
# ✅ 推荐：使用官方预训练分割模型
|
||
|
|
#model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/runs/train/seg_j/exp2/weights/best.pt')
|
||
|
|
model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/ultralytics/cfg/models/11/yolo11-seg.yaml')
|
||
|
|
# 开始训练
|
||
|
|
results = model.train(
|
||
|
|
data='/home/hx/yolo/ultralytics_yolo11-main/resize_seg_data.yaml', # 数据配置文件
|
||
|
|
epochs=1000, # 训练轮数
|
||
|
|
imgsz=640,
|
||
|
|
batch=4, # 每批图像数量
|
||
|
|
workers=10, # 数据加载线程数
|
||
|
|
device='0', # 使用 GPU 0
|
||
|
|
project='runs/train/seg_r', # 保存项目目录
|
||
|
|
name='exp', # 实验名称
|
||
|
|
exist_ok=False, # 不覆盖已有实验
|
||
|
|
optimizer='AdamW', # 可选优化器
|
||
|
|
lr0=0.001, # 初始学习率
|
||
|
|
patience=500, # 早停轮数
|
||
|
|
)
|