在这里插入图片描述

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

数据集名称 图像数量 应用方向 博客链接
🔌 电网巡检检测数据集 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张 室内场景识别 机器人导航 点击查看
🚗 汽车颜色分类数据集 2,004张 汽车颜色识别 智能交通 点击查看
🐦 鸟类物种识别数据集 9,980张 鸟类识别 生物多样性 点击查看
👨‍🦰 性别识别数据集 9,980张 性别识别 人脸识别 点击查看
⚙️ 铝材缺陷检测数据集 3,650张 智能制造 自动化检测 点击查看

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

在这里插入图片描述

🍽 食物卡路里估算数据集介绍

📌 数据集概览

本项目是专注于食物图像分类与卡路里估算的计算机视觉数据集,共包含约 6,675 张真实拍摄的食物图像,主要用于训练深度学习模型识别不同种类的餐食,并为后续的营养成分(如卡路里)估算提供基础。该数据集覆盖了常见的快餐、中式菜肴及健康食品,适用于构建智能饮食管理应用。

  • 图像数量:17741 张
  • 类别数:172类
  • 适用任务:图像分类(Image Classification)
  • 适配模型:ResNet、EfficientNet、MobileNet、ViT 等主流分类框架

包含的部分类别

类别 英文名称 描述
AW cola AW cola 常见碳酸饮料
Beijing Beef Beijing Beef 北京风味牛肉
Chow Mein Chow Mein 炒面类主食
Fried Rice Fried Rice 各式炒饭
Hashbrown Hashbrown 土豆饼
Honey Walnut Shrimp Honey Walnut Shrimp 蜜汁核桃虾
Kung Pao Chicken Kung Pao Chicken 宫保鸡丁
String Bean Chicken Breast String Bean Chicken Breast 四季豆鸡胸
Super Greens Super Greens 混合绿叶蔬菜
The Original Orange Chicken The Original Orange Chicken 经典橙子鸡
White Steamed Rice White Steamed Rice 白米饭
black pepper rice bowl black pepper rice bowl 黑椒饭
burger burger 汉堡类
carrot_eggs carrot_eggs 胡萝卜炒蛋
cheese burger cheese burger 芝士汉堡
chicken waffle chicken waffle 鸡肉华夫饼
chicken_nuggets chicken_nuggets 鸡块
chinese_cabbage chinese_cabbage 大白菜
chinese_sausage chinese_sausage 中式香肠
crispy corn crispy corn 酥脆玉米
curry curry 咖喱类菜肴
french fries french fries 薯条
fried chicken fried chicken 炸鸡
fried_dumplings fried_dumplings 煎饺
fried_eggs fried_eggs 煎蛋
mango chicken pocket mango chicken pocket 芒果鸡肉口袋饼
mozza burger mozza burger 马苏里拉汉堡
mung_bean_sprouts mung_bean_sprouts 绿豆芽
nugget nugget 鸡块
perkedel perkedel 印尼炸土豆饼
rice rice 米饭类
sprite sprite 雪碧等气泡水
tostitos cheese dip sauce tostitos cheese dip sauce 玉米片芝士蘸酱
triangle_hash_brown triangle_hash_brown 三角形土豆饼
water spinach water spinach 空心菜

数据集涵盖中西式快餐、家常菜及健康轻食,能有效支持构建面向用户的智能饮食分析与健康管理应用。

🎯 应用场景

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

  • 智能饮食记录APP
    用户拍照即可自动识别食物种类,辅助记录每日摄入热量。

  • 健康管理平台
    结合用户体重、运动量等数据,提供个性化营养建议。

  • 健身与减脂辅助工具
    帮助健身人群精确控制碳水、蛋白质和脂肪摄入。

  • 医院与营养科应用
    辅助临床营养师为患者制定膳食计划,提升效率。

  • 餐饮行业数字化
    用于餐厅菜单智能化管理,自动生成营养标签。

  • 教育与科普
    作为教学案例,帮助学生理解计算机视觉在生活中的实际应用。

🖼 数据样本展示

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

数据集包含多种真实生活场景下的食物图像:

  • 多样容器:外卖盒、盘子、碗、纸袋等不同盛装方式
  • 多角度拍摄:俯拍、斜拍、近距离特写等视角
  • 不同光照环境:自然光、室内灯光、闪光灯等条件
  • 背景复杂:桌面、餐垫、手部、餐具等干扰元素
  • 食物状态多样:完整、食用中、混合搭配等形态

图像均为真实拍摄,无合成或渲染,贴近用户日常使用场景,特别适合训练鲁棒性强的食物识别模型。

使用建议

  1. 数据预处理优化

    • 标准化图像尺寸(推荐224x224或384x384)
    • 应用数据增强:随机旋转、裁剪、亮度调整、高斯噪声
    • 可尝试归一化处理以适应不同光照条件
  2. 模型训练策略

    • 使用ImageNet预训练权重进行迁移学习
    • 尝试多模型集成(Ensemble)提升准确率
    • 对长尾类别可采用重采样或损失加权策略
  3. 实际部署考虑

    • 移动端优化:使用轻量级模型(如MobileNetV3)适配手机端
    • 实时性要求:优化推理速度,满足“拍照即识别”的用户体验
    • 离线部署:支持无网络环境下的本地识别
  4. 应用场景适配

    • APP集成:与iOS/Android应用无缝对接,提供拍照识别功能
    • 云端服务:支持API调用,供第三方平台接入
    • 智能硬件:集成到智能冰箱、料理机等设备中
  5. 性能监控与改进

    • 建立不同食物类别的混淆矩阵,针对性优化易错类别
    • 收集用户反馈样本,持续迭代模型
    • 扩展至更多菜系或细分品类(如甜点、饮品)

🌟 数据集特色

  • 真实场景采集:所有图像均来自真实用餐环境,非摆拍或合成
  • 类别丰富:覆盖36种常见食物,兼顾中西式与健康选项
  • 标注准确:由专业人员审核确认,确保类别一致性
  • 技术兼容性:支持主流分类模型及移动端部署
  • 高实用性:直接服务于健康、餐饮、教育等热门领域

📈 商业价值

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

  • 健康管理科技公司:开发智能饮食记录与营养分析产品
  • 健身APP开发商:提升用户粘性,提供精准饮食指导
  • 餐饮连锁品牌:自动生成菜品营养信息,提升品牌形象
  • 智能硬件厂商:集成到智能厨房设备中,打造智慧生活场景

🔗 技术标签

计算机视觉 图像分类 食物识别 卡路里估算 健康管理 移动应用 深度学习 营养分析 智能饮食 YOLO ResNet


注意: 本数据集适用于研究、教育和商业用途。使用时请遵守数据隐私与食品安全相关法律法规。建议在实际应用中结合营养学数据库进行卡路里与营养成分的精确推算。

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

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

更多推荐