智慧铁路异物识别 AI入门训练自己的数据集 AI识别模型学习
文章目录计算机视觉目标检测数据集项目概述核心指标数据详情训练流程一、环境准备3 行搞定二、数据集结构严格 YOLO 格式新建 my_dataset.yaml三、开始训练直接跑代码train.py四、推理流程单图/批量/视频1单张图片推理2批量图片推理3视频/摄像头推理五、关键注意点避坑六、输出目录自动生成目标检测 #YOLO格式 #计算机视觉 #开放场景感知 #智能监控数据计算机视觉目标检测数据集项目概述本数据集为通用场景下的目标检测任务构建旨在为智能感知算法的开发与验证提供高质量标注数据。数据覆盖开放场景中常见的关键目标类别可直接用于模型训练、验证与性能评估为实际应用中的目标识别任务提供数据支撑。核心指标任务类型目标检测Object Detection标注格式YOLO数据规模800张图像目标类别4类更新周期已完成稳定版本发布数据详情属性详情数据类别4个类别车辆、动物、行人、杂物数据规模800张图像应用价值1. 适用于道路、园区等开放场景的目标感知算法开发2. 可支撑智能监控、环境巡检、道路安全监测等落地场景3. 为轻量化目标检测模型的训练与优化提供数据基础。本数据集采用标准YOLO格式存储标注文件与图像一一对应可直接接入主流深度学习框架进行模型训练。所有标注均经过人工校验确保类别边界框的准确性与一致性为模型的泛化性能提供保障。训练流程下面给你一套直接可用、极简高效的 YOLO26 训练推理流程适配前面那份800 张、4 类车辆/动物/行人/杂物YOLO 格式数据集。全程基于ultralytics官方库无 NMS、端到端推理。一、环境准备3 行搞定# 安装官方库含 YOLO26pipinstallultralytics# 验证yolo version建议GPU 环境先装好CUDA PyTorch再装ultralytics。二、数据集结构严格 YOLO 格式你的数据集要长成这样my_dataset/ ├── images/ │ ├── train/ # 640 张 │ ├── val/ # 160 张 │ └── test/ # 可选 └── labels/ ├── train/ # 与图片同名 .txt ├── val/ └── test/标签.txt每行class_id x_center y_center width height归一化 0~1新建my_dataset.yamlpath:./my_datasettrain:images/trainval:images/valnc:4names:[vehicle,animal,pedestrian,misc]三、开始训练直接跑代码train.pyfromultralyticsimportYOLO# 1. 加载 YOLO26 轻量模型推荐 n/sm/l/x 更大更准modelYOLO(yolov26n.pt)# 2. 训练MuSGD 优化器、无 NMS、端到端model.train(datamy_dataset.yaml,epochs100,imgsz640,batch16,device0,# GPU 0CPU 则改为 cpuoptimizermusgd,# YOLO26 默认优化器ampTrue,# 混合精度加速workers4)# 3. 验证model.val()训练完最优权重在runs/train/exp/weights/best.pt四、推理流程单图/批量/视频1单张图片推理fromultralyticsimportYOLO modelYOLO(runs/train/exp/weights/best.pt)resmodel(test.jpg,conf0.5)res[0].show()res[0].save(result.jpg)2批量图片推理resultsmodel(sourcemy_dataset/images/val,conf0.5,saveTrue,save_txtTrue)3视频/摄像头推理resultsmodel(video.mp4,streamTrue,conf0.5)forrinresults:r.show()YOLO26无 NMS 后处理推理更快、部署更简单。五、关键注意点避坑数据集必须严格YOLO 格式坐标归一化。类别 id 从0 开始yaml 里nc4、names顺序一致。小数据集800 张epochs 80–120足够防止过拟合。推理用conf0.4~0.6平衡召回与误检。六、输出目录自动生成训练日志runs/train/exp/权重weights/best.pt/last.pt推理结果runs/detect/predict/目标检测 #YOLO格式 #计算机视觉 #开放场景感知 #智能监控数据