一、从github下载yolov11s.pt问题
官网已有反馈(可以查看这里),不是问题,只是网络问题。可以从官网下载对应的模型文件放置在训练时的根目录。
二、启动训练时卡顿在某个语句
这种情况一般会持续2到5分钟,等的真捉急,这个问题是缺少对应的字体,具体这里。
三、训练参数
results = model.train(
data="yolov11-seg.yaml", # 数据集配置文件,默认值:None
epochs=150, # 训练轮数,默认值:300
batch=8, # 批次大小,默认值:16
imgsz=1024, # 输入图像大小,默认值:640
device=0, # 训练设备 (None means automatically select, can set as '0' for GPU),默认值:None
optimizer='auto', # 优化器选择,默认值:'auto'
project='runs_yolo11_n_seg/NU', # 项目名称,保存训练结果的目录,默认值:'runs/train'
name='yolo11_1024_150', # 训练运行的名称,默认值:'exp'
save=True, # 是否保存训练结果和模型,默认值:True
save_period=-1, # 模型保存频率,-1表示保存最后的模型,默认值:-1
cache=False, # 是否缓存数据集,默认值:False
workers=8, # 数据加载线程数,默认值:8
exist_ok=False, # 是否覆盖已有项目目录,默认值:False
verbose=True, # 是否启用详细日志输出,默认值:True
seed=0, # 随机种子,默认值:0
deterministic=True, # 是否使用确定性算法,默认值:True
single_cls=False, # 是否将多类别数据集视为单一类别,默认值:False
rect=False, # 是否启用矩形训练,默认值:False
cos_lr=False, # 是否使用余弦学习率调度器,默认值:False
close_mosaic=10, # 在最后10轮禁用Mosaic数据增强,默认值:10
resume=False, # 是否从上次检查点继续训练,默认值:False
amp=True, # 是否启用自动混合精度训练,默认值:True
fraction=1.0, # 数据集的使用比例,默认值:1.0
profile=False, # 是否启用ONNX或TensorRT优化分析,默认值:False
freeze=None, # 冻结模型的前N层,默认值:None
lr0=0.01, # 初始学习率,默认值:0.01
lrf=0.01, # 最终学习率与初始学习率的比值,默认值:0.01
momentum=0.937, # 动量因子,默认值:0.937
weight_decay=0.0005, # 权重衰减,默认值:0.0005
warmup_epochs=3.0, # 预热学习率的轮次,默认值:3.0
warmup_momentum=0.8, # 预热阶段的初始动量,默认值:0.8
warmup_bias_lr=0.1, # 预热阶段的偏置学习率,默认值:0.1
box=7.5, # 边框损失的权重,默认值:7.5
cls=0.5, # 分类损失的权重,默认值:0.5
dfl=1.5, # 分布焦点损失的权重,默认值:1.5
pose=12.0, # 姿态损失的权重,默认值:12.0
kobj=1.0, # 关键点目标损失的权重,默认值:1.0
label_smoothing=0.0, # 标签平滑处理,默认值:0.0
nbs=64, # 归一化批次大小,默认值:64
overlap_mask=True, # 是否在训练期间启用掩码重叠,默认值:True
mask_ratio=4, # 掩码下采样比例,默认值:4
dropout=0.0, # 随机失活率,默认值:0.0
val=True, # 是否在训练期间启用验证,默认值:True
plots=True, # 是否生成训练曲线和验证指标图,默认值:True
# 数据增强相关参数
hsv_h=0.2, # 色相变化范围,默认值:0.2
hsv_s=0.7, # 饱和度变化范围,默认值:0.7
hsv_v=0.4, # 亮度变化范围,默认值:0.4
degrees=30.0, # 旋转角度范围,默认值:30.0
translate=0.1, # 平移范围,默认值:0.1
scale=0.5, # 缩放比例,默认值:0.5
shear=0.0, # 剪切角度范围,默认值:0.0
perspective=0.0, # 透视变化范围,默认值:0.0
flipud=0.0, # 上下翻转概率,默认值:0.0
fliplr=0.5, # 左右翻转概率,默认值:0.5
bgr=0.0, # BGR 色彩顺序调整概率,默认值:0.0
mosaic=0.5, # Mosaic 数据增强,默认值:1.0
mixup=0.0, # Mixup 数据增强,默认值:0.0
copy_paste=0.0, # Copy-Paste 数据增强,默认值:0.0
copy_paste_mode='flip', # Copy-Paste 增强模式,默认值:'flip'
auto_augment='randaugment', # 自动增强策略,默认值:'randaugment'
erasing=0.4, # 随机擦除增强比例,默认值:0.4
crop_fraction=1.0, # 裁剪比例,默认值:1.0
)