首页 > 编程语言 >ultralytics yolo 训练及推理自定义人脸关键点数据 - python 实现

ultralytics yolo 训练及推理自定义人脸关键点数据 - python 实现

时间:2024-10-20 14:21:38浏览次数:18  
标签:ultralytics img 训练 pose python yolo path 自定义

        Ultralytics YOLO 是计算机视觉和 ML 领域专业人士的高效工具

        深度学习的关键点任务也是基础和常见任务,所以这里进行基于ultralytics yolo 自定义关键点数据集的模型训练和模型推理。

        根据需求,可能会发布其 webui版本,现在已经发布了 其 检测 detect的 webui版本对应的项目地址(https://github.com/XIAN-HHappy/ultralytics-yolo-webui)。

一、安装 python 库        

安装 ultralytics 库:

pip install ultralytics 

二、数据集

该例程的数据下载地址:ultralyticsyolo训练自定义人脸关键点训练和验证数据集资源-CSDN文库

数据集信息如下:

ultralytics yolo 训练自定义人脸关键点训练和验证数据集
数据集格式:yolo
训练集数量:3295
验证集数量:120
类别:人脸,1类
类别号:0
关键点:5个,包括左眼,右眼,鼻尖,左嘴唇边界点,右嘴唇边界点。

三、人脸关键点定义

        人脸关键点示例如下图:

四、模型训练

该实例采用 yolo11-pose进行训练

4.1 定义 数据集 yaml

该数据对应的.yaml 文件命名:”yolo_face_pose_data.yaml “ 格式如下:

# Ultralytics YOLO 

标签:ultralytics,img,训练,pose,python,yolo,path,自定义
From: https://blog.csdn.net/weixin_42140236/article/details/143091313

相关文章

  • isicalendar.week——python中的WEEKNUM(ISO周数获取)
    Excel中的weeknum很方便,在python中却“没有”week_num,但可以有datetime.now().isocalendar().week。(笔记模板由python脚本于2024年10月18日11:38:15创建,本篇笔记适合喜欢python,爱鼓捣时间对象的coder翻阅)【学习的细节是欢悦的历程】Python官网:https://www.pyt......