WayBED
收藏资源简介:
WayBED数据集是一个大规模的真实世界数据集,包含来自25,353个人的超过84,000张全身照片。该数据集是作为WayBetter减肥计划的一部分收集的,用于估计身体质量指数(BMI)。数据集通过使用姿势聚类和人物检测的自动过滤方法进行筛选,以去除低质量图像。该数据集用于训练一个基于深度学习的BMI估计模型,该模型在WayBED测试集上实现了7.9%的平均绝对百分比误差(MAPE)。该模型在真实世界场景中具有广泛的应用,例如远程设置或紧急护理中,当传统方法不可用或不切实际时,可以快速评估体重。
The WayBED dataset is a large-scale real-world dataset containing over 84,000 full-body photos from 25,353 individuals. It was collected as part of the WayBetter weight loss program for body mass index (BMI) estimation. The dataset was filtered via automatic filtering methods that employ pose clustering and human detection to remove low-quality images. This dataset was used to train a deep learning-based BMI estimation model, which achieved a mean absolute percentage error (MAPE) of 7.9% on the WayBED test set. This model has broad applications in real-world scenarios such as remote settings or emergency care, enabling rapid weight assessment when traditional methods are unavailable or impractical.
Digital Scale 数据集概述
数据集基本信息
- 名称:Digital Scale
- 主要功能:基于智能手机摄像头图像进行设备端身体质量指数(BMI)估计
- 开发机构:苏黎世联邦理工学院数字健康干预中心、圣加仑大学数字健康干预中心、WayBetter公司
- 数据规模:84,963张智能手机图像(来自25,353名个体)
- 高质量数据子集:71,322张图像(经过自动过滤处理)
核心数据集详情
WayBED数据集
- 性质:大规模专有数据集
- 图像类型:智能手机拍摄的全身体图像
- 过滤方法:采用姿态聚类和人物检测的自动过滤技术,去除低质量图像(非典型姿态或不完整视图)
技术性能指标
模型性能
- WayBED测试集表现:平均绝对百分比误差(MAPE)7.9%
- VisualBodyToBMI数据集表现:MAPE 13%(训练期间完全未见的数据集)
- 泛化能力:与在该数据集上训练的最先进方法相当
部署与应用
- 部署框架:CLAID框架(用于Android设备部署)
- 功能范围:完整管道部署,包括图像过滤和BMI估计
- 实时能力:支持现代GPU上的实时推理
- 兼容性:支持无GPU环境的CPU兼容
代码资源结构
主要目录
app/:Android应用实现(使用CLAID进行实时BMI估计)docs/:文档文件get_started/:快速入门指南(最小化BMI预测设置)src/:源代码src/scripts:训练、评估和部署脚本trained_models/keypoint_models/:预训练关键点模型(用于姿态聚类)
核心脚本分类
模型训练与部署
- 训练脚本:支持并行GPU训练和不同模型架构
- 评估脚本:多模型评估和系统性能分析
- 部署脚本:使用CLAID框架的完整BMI估计管道
预处理与过滤
- 姿态聚类训练:使用关键点数据训练聚类模型
- 姿态预测:从原始关键点数据预测姿态聚类
- 边界框检测:人物区域识别和质量评估
- 数据集解析:从文件名提取元数据(体重、身高、性别)
使用要求
- 模型权重:需联系作者获取(rmanichand@ethz.ch 或 planger@ethz.ch)
- 依赖安装:通过pip安装requirements.txt指定依赖
- 快速开始:提供get_started文件夹简化设置流程
开源贡献
- 完整代码开源:包括模型训练、过滤和移动部署的CLAID包
- 预训练模型:提供关键点模型用于姿态聚类
- 文档资料:包含完整设置说明和故障排除指南




