kaggle中的2D目标检测训练trick总结
·
数据增强
- 主要是mixup和cutmix,详见https://docs.ultralytics.com/zh/guides/yolo-data-augmentation/#copy-paste-mode-copy_paste_mode
你可以在yolo中这样使用
model = YOLO("yolo11m.pt")
data_yaml = "/kaggle/working/yolo_params.yaml"
model.train(
data=data_yaml,
epochs=10,
batch=4,
imgsz=1056,
patience=150,
optimizer='SGD',
momentum=0.937,
lr0=0.001,
weight_decay=0.0005,
cos_lr=True,
save_period=1,
workers=8,
# Augmentations
close_mosaic=20,
hsv_h=0.015,
hsv_s=0.7,
hsv_v=0.4,
flipud=0.5,
fliplr=0.5,
translate=0.1,
scale=0.5,
shear=0.01
)
模型训练
后处理
DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐

所有评论(0)