Multi-Modal-and-Distributed-mmWave-ISAC-Datasets-for-Human-Sensing
收藏资源简介:
该存储库提供了一个基于PyTorch的框架,用于在各种毫米波ISAC数据集上训练和评估深度学习模型,包括手势识别、姿态估计、定位和步态识别。它支持分类和回归任务,并在适用时包括背景减除选项。支持多个数据集:mmWGesture – 毫米波手势识别(分类)、5GmmGesture – 5G毫米波手势识别(分类)、mmWPose – 毫米波骨骼姿态估计(回归)、DISAC-mmVRPose – 基于VR的毫米波姿态估计(回归)、mmW-Loc – 毫米波定位(带可选背景减除,分类)、mmW-GaitID – 毫米波步态识别(带可选背景减除,分类)。
This repository provides a PyTorch-based framework for training and evaluating deep learning models across various millimeter-wave ISAC datasets, including gesture recognition, pose estimation, localization, and gait recognition. It supports both classification and regression tasks, with optional background subtraction where applicable, and encompasses multiple datasets: mmWGesture – millimeter-wave gesture recognition (classification), 5GmmGesture – 5G millimeter-wave gesture recognition (classification), mmWPose – millimeter-wave skeletal pose estimation (regression), DISAC-mmVRPose – VR-based millimeter-wave pose estimation (regression), mmW-Loc – millimeter-wave localization (with optional background subtraction, classification), and mmW-GaitID – millimeter-wave gait recognition (with optional background subtraction, classification).
数据集概述
数据集名称
Multi-Modal-and-Distributed-mmWave-ISAC-Datasets-for-Human-Sensing
数据集类型
多模态分布式毫米波ISAC(集成传感与通信)数据集
主要用途
人类感知任务,包括手势识别、姿态估计、定位和步态识别
支持的数据集
- mmWGesture:毫米波手势识别(分类任务)
- 5GmmGesture:5G毫米波手势识别(分类任务)
- mmWPose:毫米波骨骼姿态估计(回归任务)
- DISAC-mmVRPose:基于VR的毫米波姿态估计(回归任务)
- mmW-Loc:毫米波定位(分类任务,支持可选背景减除)
- mmW-GaitID:毫米波步态识别(分类任务,支持可选背景减除)
框架特性
- 基于PyTorch的深度学习框架
- 支持分类和回归任务
- 通用ResNet18架构,具有灵活的输入通道
- 支持交叉熵损失(CrossEntropyLoss)和均方误差损失(MSELoss)
使用方式
通过运行python main.py启动训练和评估




