技术栈:YOLOv8、PyTorch、COCO数据集格式、数据增强
基于YOLOv8框架完成目标检测全流程实践,使用船舶数据集共7000张图像,按7:2:1划分训练集、验证集、测试集;
适配COCO数据集格式,调整YOLOv8官方配置文件(yolov8s.yaml和coco.yaml),优化超参数(如imgsz=640、batch_size=64);
应用默认数据增强策略(如Mosaic、随机翻转、色彩空间变换),训练后模型在测试集达到mAP50:98%、Recall:95%;
使用Gradio搭建简易推理Demo,实现本地图像/视频流检测可视化。
点击空白处退出提示












评论