在这里插入图片描述

📦 已发布目标检测数据集合集(持续更新)

数据集名称 图像数量 应用方向 博客链接
🔌 电网巡检检测数据集 1600 张 电力设备目标检测 点击查看
🔥 火焰 / 烟雾 / 人检测数据集 10000张 安防监控,多目标检测 点击查看
🚗 高质量车牌识别数据集 10,000 张 交通监控 / 车牌识别 点击查看
🌿 农田杂草航拍检测数据集 1,200 张 农业智能巡检 点击查看
🐑 航拍绵羊检测数据集 1,700 张 畜牧监控 / 航拍检测 点击查看
🌡️ 热成像人体检测数据集 15,000 张 热成像下的行人检测 点击查看
🦺 安全背心检测数据集 3,897 张 工地安全 / PPE识别 点击查看
🚀 火箭检测数据集介绍 12,000 张 智慧医疗 / 养老护理 点击查看
⚡ 绝缘子故障检测数据集 2,100张 无人机巡检/智能运维 点击查看
🚦交通标志检测数据集 1866张 智能驾驶系统/地图数据更新 点击查看
🚧 道路交通标志检测数据集 2,000张 智能地图与导航/交通监控与执法 点击查看
😷 口罩检测数据集 1,600张 疫情防控管理/智能门禁系统 点击查看
🦌 野生动物检测数据集 5,138张 野生动物保护监测/智能狩猎相机系统 点击查看
🍎 水果识别数据集 2,611张 图片智能零售/智慧农业 点击查看
🚁 无人机目标检测数据集 14,751张 无人机检测/航拍图像 点击查看
🚬 吸烟行为检测数据集 2,108张 公共场所禁烟监控/健康行为研究 点击查看
🛣️ 道路坑洞检测数据集 8,300张 智能道路巡检系统/车载安全监测设备 点击查看
🛠️ 井盖识别数据集 2,700 张 道路巡检 智能城市 点击查看
🧯 消防器材检测数据集 9,600 张 智慧安防系统 自动审核系统 点击查看
📱 手机通话检测数据集 3,100张 智能监控系统 驾驶安全监控 点击查看
🚜 建筑工地车辆检测数据集 28,000 张 施工现场安全监控 智能工地管理系统 点击查看
🏊 游泳人员检测数据集 4,500 张 游泳池安全监控 海滩救生系统 点击查看
🌿 植物病害检测数据集 6,200 张 智能农业监测系统 家庭园艺助手 点击查看
🐦 鸟类计算机视觉数据集 6,200 张 鸟类保护监测 生态环境评估 点击查看
🚁 无人机计算机视觉数据集 7,000 张 空域安全监管 无人机反制系统 点击查看
🛡️ Aerial_Tank_Images 坦克目标检测数据集 2,200 张 军事目标识别与侦查 卫星遥感目标识别 点击查看
♻️ 塑料可回收物检测数据集 10,000 张 智能垃圾分类系统 环保回收自动化 点击查看
🏢 建筑物实例分割数据集 9,700 张 城市规划与发展 智慧城市管理 点击查看
😊 人脸情绪检测数据集 9,400 张 智能客服系统 在线教育平台 点击查看
🔍 红外人员车辆检测数据集 53,000 张 智能安防监控系统 边境安全防控 点击查看
🚗 停车空间检测数据集 3,100 张 实时车位导航系统 智能停车收费管理 点击查看
♻ 垃圾分类检测数据集 15,000 张 智能垃圾分类 回收站与环保设施自动化 点击查看
✂️ 石头剪刀布手势识别数据集 3,100 张 智能游戏系统 人机交互界面 点击查看
🍌 腐烂香蕉检测数据集 4,267张 食品质量检测 智能农产品分拣系统 点击查看
🎰 扑克牌数字检测数据集 6,240 张 智能扑克游戏系统 赌场监控与安全 点击查看
🚗 车牌识别数据集 12,658张 智能交通管理系统 停车场自动化管理 点击查看
🏗️ 建筑设备检测数据集 6,247张 智能工地管理 施工安全监控 点击查看
🦺 个人防护装备检测数据集 7,892 张 工业安全监控 建筑工地安全管理 点击查看
⚓ 船舶检测数据集 7,542张 海洋交通监管 港口智能化管理 点击查看
🚁 空中救援任务数据集 6,742张 自然灾害应急救援 海上搜救任务 点击查看
✈️ 固定翼无人机检测数据集 8,247张 空域安全监管 机场反无人机系统 点击查看
😷 口罩检测数据集 8,432张 公共场所监控系统 企业复工防疫管理 点击查看
🚁 无人机检测数据集 6,847张 机场空域安全管理 重要设施防护监控 点击查看
✂️ 剪刀石头布手势识别数据集 2,376张 智能游戏开发 儿童教育娱乐 点击查看
🦺 安全背心识别数据集 4,892张 建筑工地安全监管 工业园区智能巡检 点击查看
🥤 饮料容器材质检测数据集 6,342张 智能垃圾分拣系统 生产线质量检测 点击查看
🚚 物流运输场景数据集 7,854张 智能仓储管理系统 物流车队智能调度 点击查看
🌡️ 热成像数据集 9,127张 夜间安防监控 工业设备检测 点击查看
🚗 车辆损伤识别数据集 6,742 张 保险理赔自动化 智能汽车维修评估 点击查看
🃏 扑克牌牌面识别数据集 8,432 张 智能扑克游戏系统 线上扑克直播辅助 点击查看
🔴 围棋棋子检测数据集 8,247 张 智能围棋对弈系统 围棋教学平台 点击查看
🚀 火箭检测数据集 6,425 张 航天发射监测 军事情报分析 点击查看
⚡ 摔跤跌倒检测数据集 9,354 张 体育安全监测系统 智能运动防护设备 点击查看
🚗 PKLot停车位检测数据集 12,416 张 计算机视觉 停车位检测 点击查看
🚗 车辆分类数据集 28,045 张 车辆识别 交通工具 点击查看
🚦 道路标识检测数据集 2,893 张 道路标识识别 自动驾驶 点击查看
📦 集装箱侧面分类数据集 2,408 张 集装箱识别 港口物流 点击查看
🚦 交通与道路标识检测数据集 10,000张 交通标志识别 自动驾驶 点击查看
🎯 COCO数据集 123,272张 目标检测 COCO 点击查看
👥 人群检测数据集 7,300张 人流统计 行人检测 点击查看
🔢 MNIST手写数字识别数据集 70,000张 图像分类 手写识别 点击查看
🐦 鸟类物种识别数据集 9,880张 鸟类识别 生态保护 点击查看
🩺 皮肤癌检测数据集 9,900张 皮肤癌检测 医学影像 点击查看
🚗 汽车颜色分类数据集 2,004张 汽车识别 颜色检测 点击查看
⚔️ 暴力与非暴力行为识别数据集 10,000张 行为识别 暴力检测 点击查看
🌿 植物病害检测数据集 5,500张 农业AI 植物病害识别 点击查看
🧠 脑肿瘤检测数据集 9,900张 医学影像 脑肿瘤识别 点击查看
🏀 篮球场景目标检测数据集 4,100张 体育AI 篮球分析 点击查看
⚽ 足球场景目标检测数据集 6,700张 体育AI 足球分析 点击查看
🗑️ 垃圾分类检测数据集 10,464张 垃圾分类 环保科技 点击查看
🚁 无人机检测数据集 9,495张 无人机识别 低空安全 点击查看
😊 人类面部情绪识别数据集 9,400张 情绪识别 人脸识别 点击查看
🔥 烟雾与火灾检测数据集 536张 火灾检测 烟雾识别 点击查看
🔥 火灾检测计算机视觉数据集 10,967张 火灾检测 火灾预警 点击查看
🌐 网站截图计算机视觉数据集 1,286张 网页分析 UI自动化 点击查看
🛣️ 车道线实例分割数据集 1,610张 车道线检测 自动驾驶 点击查看
🛣️ 道路实例分割数据集 1,114张 实例分割 道路检测 点击查看
🚗 汽车损伤检测数据集 4500张 汽车损伤识别 保险定损 点击查看
🏗️ 建筑物实例分割数据集 9,700张 遥感图像 建筑物提取 点击查看
🥚 CVR EGG 实例分割数据集 1,438张 禽蛋检测 农业AI 点击查看
🚪 房间检测计算机视觉数据集 1,272张 实例分割 建筑图纸识别 点击查看
💅 美甲实例分割数据集 3,626张 美甲识别 虚拟试妆 点击查看
🚗 汽车损伤严重程度分割数据集 2,485张 汽车损伤检测 保险定损 点击查看
🪵 木材缺陷检测数据集 10,000张 木材缺陷检测 工业质检 点击查看
🧑‍🦯 人体姿态与行为实例分割数据集 4,567张 人体姿态识别 行为分析 点击查看
📦 条形码检测数据集 9,988张 条形码识别 零售自动化 点击查看
🚗 道路车辆检测数据集 4,058张 自动驾驶 车辆识别 点击查看
🎮 麻将计算机视觉模型数据集 212张 麻将识别 游戏AI 点击查看
🛡️ 个人防护装备检测数据集 12,879张 安全生产 工业AI 点击查看
🅰️ OCR字符检测数据集 12,879张 OCR字符检测 车牌识别 点击查看
🔫 武器检测数据集 9,672 张 武器识别 公共安全 点击查看
🔥 火灾检测数据集 8,939 张 火灾识别 消防安全 点击查看
🧱 墙体检测计算机视觉数据集 6,646 张 墙体识别 建筑图纸解析 点击查看
🩸 肝病细胞检测数据集 105 张 细胞识别 数字病理 点击查看
🚗 CCTV车辆与摩托车检测数据集 1,023 张 车辆识别 摩托车检测 点击查看
🍅 番茄叶片病害检测数据集 4,132 张 植物病害识别 智慧农业 点击查看
🔥 火灾与烟雾检测数据集 8,875 张 火灾识别 烟雾检测 点击查看
🎮 CSGO 游戏目标检测数据集 2,427张 游戏AI CSGO 点击查看
🚬 吸烟行为检测数据集 3,895张 吸烟行为识别 公共健康 点击查看
🔪 刀具检测数据集 9,219张 刀具识别 枪械检测 点击查看
🐾 动物目标检测数据集 1,000张 动物识别 智能农场 点击查看
🃏 扑克牌检测数据集 1,300张 扑克牌识别 游戏AI 点击查看
🚨 跌倒检测数据集 4,600张 跌倒检测 行为识别 点击查看
🛡️ 军用车辆检测数据集 3,143张 军用车辆识别 战场感知 点击查看
🔧 电缆损伤检测数据集 1,318张 电缆损伤识别 工业质检 点击查看
👤 人物检测数据集 1,687张 人物识别 安防监控 点击查看
🛡️ 军事目标检测数据集 6,149张 军事识别 无人机侦察 点击查看
🚀 火箭检测计算机视觉数据集 12,303张 火箭识别 航天监控 点击查看
🏗️ 建筑工地PPE检测数据集 8,845张 PPE识别 工地安全 点击查看
👤 人物检测计算机视觉数据集 2,545张 人物检测 安防监控 点击查看
📱 驾驶员行为检测数据集 8,867张 人物检测 安防监控 点击查看
🌙 红外行人与车辆检测数据集 53,483张 红外成像 行人检测 点击查看
🏐 排球动作检测数据集 13,925张 排球动作识别 体育分析 点击查看
🗑️ 水域垃圾检测数据集 2,273张 水域垃圾识别 环保监测 点击查看
🚗 达卡城市交通目标检测数据集 1,502张 城市交通 达卡数据集 点击查看
⚙️ 金属结构腐蚀检测数据集 1,249张 工业缺陷检测 腐蚀识别 点击查看
🚦 交通标志检测数据集 4,113张 交通标志识别 自动驾驶 点击查看
🅿️ 停车位状态检测数据集 3,123张 智能停车 车位识别 点击查看
⛳ 高尔夫球与球杆检测数据集 6,082张 高尔夫分析 运动科学 点击查看
🖥️ UI元素检测数据集 5,428张 UI自动化 无障碍访问 点击查看
✋ 手势识别数据集 2,122张 手势识别 人机交互 点击查看
🛒 杂货商品检测数据集 83,699张 商品识别 智能零售 点击查看
📷 野外相机动物检测数据集 1,311张 野外相机 野生动物识别 点击查看
🚜 工程机械检测数据集 2,655张 工程机械识别 智慧工地 点击查看
⚽ 足球检测数据集 1,237张 足球识别 体育分析 点击查看
🏀 篮球运动目标检测数据集 3,666张 篮球识别 体育分析 点击查看
🚧 障碍物检测数据集 9,183张 障碍物识别 自动驾驶 点击查看
⚠️ 安全锥检测数据集 1,703张 安全锥识别 自动驾驶 点击查看
♟ 国际象棋棋子检测数据集 3,946张 棋子识别 国际象棋 点击查看
👤 人体检测数据集 7,785张 人体识别 行人检测 点击查看
🩻 X光手部骨骼检测数据集 3,839张 医学影像 X光识别 点击查看
🛒 R2P2 食品杂货检测数据集 2,745张 食品识别 智能零售 点击查看
🛋️ 室内家具检测数据集 8,055张 室内设计 智能家居 点击查看
🏗️ 建筑工程车辆检测数据集 7,615张 智慧工地 施工安全 点击查看
🎥 航拍军事目标检测数据集 10,000张 军事识别 无人机侦察 点击查看
🔥 火灾检测数据集 86,617张 火灾识别 烟雾检测 点击查看
💥 暴力与武器检测数据集 5,953张 暴力行为识别 武器检测 点击查看
🐾 牛津宠物数据集 3,680张 宠物识别 细粒度分类 点击查看
🛒 超市货架空位检测数据集 1,444张 货架空位检测 缺货识别 点击查看
🚧 街道无障碍设施检测数据集 4,968张 无障碍设施 智慧城市 点击查看
🎾 网球检测数据集 2,244张 网球识别 体育分析 点击查看
🚁 无人机检测数据集 7,248张 无人机识别 空域安全 点击查看
🤖 机器人视觉垃圾分类数据集 7,984张 垃圾分类 智能机器人 点击查看
🐕 斯坦福犬类数据集 9,984张 犬种识别 细粒度分类 点击查看
🍎 水果检测数据集 1,007张 水果识别 智能零售 点击查看
🔥 火源检测数据集 9,128张 火源识别 智能安防 点击查看
👷 个人防护装备检测数据集 3,551张 个人防护装备 智慧工地 点击查看
👤 人体检测数据集 10,000张 人体检测 智能监控 点击查看
🦁 多物种动物检测数据集 9,073张 野生动物识别 生态保护 点击查看
🐱 猫只检测数据集 1,159张 猫只识别 宠物管理 点击查看
🐷 猪只检测数据集 1,092张 猪只识别 智慧养殖 点击查看
🗑️ 垃圾分类与物体检测数据集 2,362张 垃圾分类 环保科技 点击查看
🖐️ 印度手语检测数据集 1,748张 手语识别 无障碍沟通 点击查看
⚽ 足球比赛分析数据集 8,873张 足球分析 体育科技 点击查看
🍅 番茄叶片病害检测数据集 8,439张 植物病害 智慧农业 点击查看
🛡️ MilTech 军事目标检测数据集 4,690张 军事目标识别 战场感知 点击查看
🧭 仪表盘指针检测数据集 4,862张 工业仪表识别 指针检测 点击查看
👤 COCO 人物检测数据集 5,438张 人物检测 COCO 点击查看
🚜 挖掘机与工程车辆检测数据集 2,655张 工程车辆识别 智慧工地 点击查看
✋ 美国手语字母检测数据集 720张 手语识别 美国手语 点击查看
🍌 香蕉成熟度分类数据集 5,616张 图像分类 香蕉成熟度 点击查看

📌 每篇文章附带模型指标、训练思路与推理部署建议,欢迎点赞收藏支持~

在这里插入图片描述

🏢 MIT室内场景识别数据集介绍

📌 数据集概览

本项目是专注于室内环境场景分类的计算机视觉数据集,共包含约 15,571 张真实拍摄图像,主要用于训练深度学习模型对不同功能的室内空间进行精准识别与分类。该数据集是构建智能建筑、机器人导航、虚拟现实和安防监控系统的核心基础。

  • 图像数量:15,571 张
  • 类别数:67 类
  • 适用任务:图像分类(Image Classification)
  • 适配模型:ResNet、VGG、EfficientNet、Vision Transformer (ViT) 等主流分类网络

包含类别

类别 英文名称 描述
机场内部 airport_inside 机场候机厅、登机口等区域
艺术工作室 artstudio 画家或设计师的工作空间
礼堂 auditorium 大型会议或演出场所
面包店 bakery 售卖面包糕点的店铺
酒吧 bar 提供酒水服务的休闲场所
浴室 bathroom 家庭或公共卫生间
卧室 bedroom 休息睡眠的空间
书店 bookstore 销售书籍的零售空间
保龄球馆 bowling 进行保龄球运动的场馆
自助餐区 buffet 提供自助餐饮服务的区域
赌场 casino 提供赌博娱乐的场所
儿童房 children_room 专为儿童设计的房间
教堂内部 church_inside 宗教活动场所内部
教室 classroom 学校教学空间
衣帽间 closet 存放衣物的储物空间
服装店 clothingstore 销售服装的零售店
计算机房 computerroom 配备多台电脑的办公或教学空间
音乐厅 concert_hall 举办音乐会的大型场馆
走廊 corridor 连接不同房间的通道
熟食店 deli 销售熟食和三明治的店铺
牙科诊所 dentaloffice 牙科诊疗工作区域
餐厅 dining_room 用餐的主要空间

数据集覆盖了从住宅到商业、从教育到娱乐的67种典型室内场景,能够显著提升模型在复杂室内环境下的语义理解能力。

🎯 应用场景

该数据集非常适用于以下场景与研究方向:

  • 智能建筑与家居自动化
    根据房间类型自动调节灯光、温度和设备运行模式,实现真正的智能家居。

  • 服务机器人导航
    帮助机器人识别当前所处环境(如“厨房”、“走廊”),规划路径并执行相应任务。

  • 虚拟现实与游戏开发
    自动生成符合场景逻辑的虚拟室内环境,提升沉浸式体验的真实性。

  • 安防与监控系统
    自动识别监控画面中的地点(如“银行大厅”、“酒店前台”),辅助事件分析与预警。

  • 房地产与室内设计
    快速分类和归档房屋照片,用于在线平台展示或设计方案推荐。

  • 增强现实(AR)应用
    在手机或AR眼镜中实时识别用户所处室内环境,提供位置相关的交互信息。

🖼 数据样本展示

以下展示部分数据集内的样本图片:
在这里插入图片描述
在这里插入图片描述

数据集包含多种真实室内环境下的图像:

  • 多样化场景:涵盖住宅、商业、教育、交通、娱乐等六大类空间
  • 多视角拍摄:广角、俯视、平视等多种角度,模拟真实感知
  • 光照变化:自然光、人工照明、混合光源等不同条件
  • 家具与陈设:包含不同风格、年代和文化背景的室内布置
  • 高分辨率:清晰呈现墙面、地板、天花板、装饰物等细节特征

图像采集于全球多个城市的真实室内环境,数据多样性优秀,特别适合训练鲁棒性强的场景识别模型。

使用建议

  1. 数据预处理优化

    • 统一图像尺寸(推荐224x224或384x384)
    • 应用标准化归一化(如ImageNet均值方差)
    • 对小样本类别(如“cloister”、“concert_hall”)进行过采样或数据增强
  2. 模型训练策略

    • 使用迁移学习,在ImageNet预训练模型基础上微调
    • 采用交叉验证确保模型泛化能力
    • 对易混淆类别(如“bedroom”与“hotel_room”,“classroom”与“lecture_hall”)进行重点强化
  3. 实际部署考虑

    • 边缘设备优化:使用轻量级模型(如MobileNetV3)部署于机器人或IoT设备
    • 实时推理速度:优化模型以满足导航或监控系统的实时性需求
    • 低功耗设计:适配电池供电或嵌入式设备运行环境
  4. 应用场景适配

    • 机器人集成:与SLAM系统结合,实现语义地图构建
    • 移动端部署:支持智能手机现场拍摄识别
    • 云端API服务:提供批量上传与分析接口,服务于大型地产或安防企业
  5. 性能监控与改进

    • 建立不同光照、遮挡、视角下的准确率基准测试
    • 收集误分类样本(如“bar”与“restaurant”混淆)进行模型强化
    • 定期更新数据集以涵盖更多新兴场景(如“电竞馆”、“共享办公”)

🌟 数据集特色

  • 高质量标注:由计算机视觉专家和场景设计师共同定义并审核类别
  • 场景覆盖广:完整涵盖日常生活中可能遇到的绝大多数室内空间
  • 学术影响力大:作为经典基准数据集被广泛应用于CVPR、ICCV等顶级会议论文
  • 技术兼容性:支持主流深度学习框架和部署平台
  • 持续更新:计划增加新类别和更高分辨率图像

📈 商业价值

该数据集在以下商业领域具有重要价值:

  • 智能硬件制造商:提升扫地机器人、服务机器人的环境认知能力
  • 房地产科技公司:自动化房屋照片分类与价值评估
  • 安防系统供应商:增强监控视频的语义理解与智能告警
  • AR/VR内容开发商:快速生成符合场景逻辑的虚拟空间

🔗 技术标签

计算机视觉 图像分类 室内场景识别 机器人导航 智能建筑 深度学习 机器学习 语义理解 安防监控 虚拟现实


注意: 本数据集适用于研究、教育和商业用途。在实际应用中,建议结合具体业务场景对模型输出进行后处理,并考虑遮挡、低光照等现实因素对识别精度的影响。

YOLOv8 训练实战

本教程介绍如何使用 YOLOv8 对目标进行识别与检测。涵盖环境配置、数据准备、训练模型、模型推理和部署等全过程。


📦 1. 环境配置

建议使用 Python 3.8+,并确保支持 CUDA 的 GPU 环境。

# 创建并激活虚拟环境(可选)
python -m venv yolov8_env
source yolov8_env/bin/activate  # Windows 用户使用 yolov8_env\Scripts\activate

安装 YOLOv8 官方库 ultralytics

pip install ultralytics

📁 2. 数据准备

2.1 数据标注格式(YOLO)

每张图像对应一个 .txt 文件,每行代表一个目标,格式如下:

<class_id> <x_center> <y_center> <width> <height>

所有值为相对比例(0~1)。

类别编号从 0 开始。

2.2 文件结构示例

datasets/
├── images/
│   ├── train/
│   └── val/
├── labels/
│   ├── train/
│   └── val/

2.3 创建 data.yaml 配置文件

path: ./datasets
train: images/train
val: images/val

nc: 11
names: ['Bent_Insulator', 'Broken_Insulator_Cap', '', ...]

🚀 3. 模型训练

YOLOv8 提供多种模型:yolov8n, yolov8s, yolov8m, yolov8l, yolov8x。可根据设备性能选择。

yolo detect train \
  model=yolov8s.pt \
  data=./data.yaml \
  imgsz=640 \
  epochs=50 \
  batch=16 \
  project=weed_detection \
  name=yolov8s_crop_weed
参数 类型 默认值 说明
model 字符串 - 指定基础模型架构文件或预训练权重文件路径(.pt/.yaml
data 字符串 - 数据集配置文件路径(YAML 格式),包含训练/验证路径和类别定义
imgsz 整数 640 输入图像的尺寸(像素),推荐正方形尺寸(如 640x640)
epochs 整数 100 训练总轮次,50 表示整个数据集会被迭代 50 次
batch 整数 16 每个批次的样本数量,值越大需要越多显存
project 字符串 - 项目根目录名称,所有输出文件(权重/日志等)将保存在此目录下
name 字符串 - 实验名称,用于在项目目录下创建子文件夹存放本次训练结果

关键参数补充说明:

  1. model=yolov8s.pt

    • 使用预训练的 YOLOv8 small 版本(平衡速度与精度)
    • 可用选项:yolov8n.pt(nano)/yolov8m.pt(medium)/yolov8l.pt(large)
  2. data=./data.yaml

    # 典型 data.yaml 结构示例
    path: ../datasets/weeds
    train: images/train
    val: images/val
    names:
      0: Bent_Insulator
      1: Broken_Insulator_Cap
      2: ...
      3: ...
    

📈 4. 模型验证与测试

4.1 验证模型性能

yolo detect val \
  model=runs/detect/yolov8s_crop_weed/weights/best.pt \
  data=./data.yaml
参数 类型 必需 说明
model 字符串 要验证的模型权重路径(通常为训练生成的 best.ptlast.pt
data 字符串 与训练时相同的 YAML 配置文件路径,需包含验证集路径和类别定义

关键参数详解

  1. model=runs/detect/yolov8s_crop_weed/weights/best.pt

    • 使用训练过程中在验证集表现最好的模型权重(best.pt
    • 替代选项:last.pt(最终epoch的权重)
    • 路径结构说明:
      runs/detect/
      └── [训练任务名称]/
          └── weights/
              ├── best.pt   # 验证指标最优的模型
              └── last.pt   # 最后一个epoch的模型
      
  2. data=./data.yaml

    • 必须与训练时使用的配置文件一致
    • 确保验证集路径正确:
      val: images/val  # 验证集图片路径
      names:
        0: crop
        1: weed
      

常用可选参数

参数 示例值 作用
batch 16 验证时的批次大小
imgsz 640 输入图像尺寸(需与训练一致)
conf 0.25 置信度阈值(0-1)
iou 0.7 NMS的IoU阈值
device 0/cpu 选择计算设备
save_json True 保存结果为JSON文件

典型输出指标

Class     Images  Instances      P      R      mAP50  mAP50-95
all        100       752      0.891  0.867    0.904    0.672
crop       100       412      0.912  0.901    0.927    0.701
weed       100       340      0.870  0.833    0.881    0.643

4.2 推理测试图像

yolo detect predict \
  model=runs/detect/yolov8s_crop_weed/weights/best.pt \
  source=./datasets/images/val \
  save=True

🧠 5. 自定义推理脚本(Python)

from ultralytics import YOLO
import cv2

# 加载模型
model = YOLO('runs/detect/yolov8s_crop_weed/weights/best.pt')

# 推理图像
results = model('test.jpg')

# 可视化并保存结果
results[0].show()
results[0].save(filename='result.jpg')

🛠 6. 部署建议

✅ 本地运行:通过 Python 脚本直接推理。

🌐 Web API:可用 Flask/FastAPI 搭建检测接口。

📦 边缘部署:YOLOv8 支持导出为 ONNX,便于在 Jetson、RKNN 等平台上部署。

导出示例:

yolo export model=best.pt format=onnx

📌 总结流程

阶段 内容
✅ 环境配置 安装 ultralytics, PyTorch 等依赖
✅ 数据准备 标注图片、组织数据集结构、配置 YAML
✅ 模型训练 使用命令行开始训练 YOLOv8 模型
✅ 验证评估 检查模型准确率、mAP 等性能指标
✅ 推理测试 运行模型检测实际图像目标
✅ 高级部署 导出模型,部署到 Web 或边缘设备
Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐