reRLDD 和 reDROZY
收藏MDFG: Multi-Dimensional Fine-Grained Modeling for Fatigue Detection
数据集
本实验使用了 reRLDD 和 reDROZY 数据集。
数据集统计
数据集标签的统计结果存放在 ./Label_statistics 目录中。
数据集设置
-
训练和测试集划分:
- 前8名参与者作为训练集。
- 剩余的参与者作为测试集。
-
特征提取:
- 使用眼动特征比率(EAR)输入到小波包中进行特征提取。
- 提取的特征存储在
./wavelet_feature目录中。
-
可信样本选择:
-
使用
small_loss方法识别可信样本。 -
更新路径:
path="your_wavelet_feature_path"和save_path="your_confident_data_save_path"。 -
运行以下命令提取可信样本: python python ./FGT/small_loss_confident_ind.py --dataset reRLDD
-
可信样本和剩余样本以及归一化后的数据集存储在
./confident_analysis目录中。
-
-
细粒度特征提取:
-
将可信样本、训练集和测试集输入到
gmm_extract_Fine_feature.py中进行细粒度分类。 -
更新路径:
path="your_confident_data_save_path"和save_path="your_fine_grained_data_save_path"。 -
运行以下命令生成细粒度样本: python python ./gmm_extract_Fine_feature.py --dataset reRLDD
-
生成的量化疲劳序列特征存储在
./fine_grained_feature目录中。
-
训练与测试
-
不使用元学习: python python ./MDFG_wometa.py --dataset reRLDD
-
使用元学习: python python ./MDFG_meta.py --dataset reRLDD
重要提示:在运行这些命令之前,请确保修改细粒度数据和相应标签路径。




