HAR-Dataset-Preprocess
HAR-Dataset-Preprocess copied to clipboard
xushige
trafficstars
Human Activity Recogniton (HAR) 公开数据集预处理与网络搭建
下载工程
git clone https://github.com/xushige/HAR-Dataset-Preprocess.git
安装库依赖
cd HAR-Dataset-Prerocess
pip3 install -r requirements.txt
模型训练代码运行样例【或者直接编译器运行train.py文件，在文件中修改参数:--dataset, --model】
python3 train.py --dataset unimib --model vit
--dataset choose from 【'uci', 'unimib', 'usc', 'pamap', 'wisdm', 'dasa', 'oppo'】
--model choose from 【'cnn', 'resnet', 'res2net', 'resnext', 'sknet', 'resnest', 'lstm', 'ca', 'sa', 'dilation', 'depthwise', 'shufflenet', 'dcn', 'vit', 'swin'】
包含数据集
- [x] Daily-and-Sports-Activities
- [x] PAMAP2
- [x] UCI-HAR
- [x] USC-HAD
- [x] UniMiB-SHAR
- [x] WISDM
- [x] OPPORTUNITY
包含模型
- [x] 'cnn': cnn.CNN
- [x] 'resnet': resnet.ResNet
- [x] 'res2net': res2net.Res2Net
- [x] 'resnext': resnext.ResNext
- [x] 'sknet': sk_resnet.SKResNet
- [x] 'resnest': resnest.ResNeSt
- [x] 'lstm': lstm.LSTM
- [x] 'ca': channel_attention.ChannelAttentionNeuralNetwork
- [x] 'sa': spatial_attention.SpatialAttentionNeuralNetwork
- [x] 'dilation': dilated_conv.DilatedConv
- [x] 'depthwise': depthwise_conv.DepthwiseConv
- [x] 'shufflenet': shufflenet.ShuffleNet
- [x] 'dcn': dcn.DeformableConvolutionalNetwork
- [x] 'vit': vit.VisionTransformer
- [x] 'swin': swin.SwinTransformer
仅预处理代码运行样例, 这里以WISDM数据集为例【或者直接编译器运行train.py文件，在文件中修改参数】
python3 WISDM/dataproc.py
xushige