This repository contains the code used to preprocess the FatigueSet dataset that comes from the study [1], and was produced as a part of the research project "Early Fadigue Detection using Wearables as a means for the Reduction of Critical Accidents in the Mining Process", which was developed by a team of our lab during our participation in the AI League of the 2025 Pan-African Robotics Competition.
The data preprocessing steps tries to follow most of the steps presented in the study [2], but with some modifications according to our needs.
The preprocessed data can be found here and has resulted in the following notebook in which we used the preprocessed data to create a LSTM Fatigue Detection Classifier Meta Learning Model using the concepts of "Continuous Meta-Learning without Tasks" presented in the study [3].
-
Download the fatigueset dataset from it's repository:
curl https://sensix.tech/datasets/fatigueset/fatigueset.zip -o fatigueset.zip
1.1. Unzip the dataset:
unzip fatigueset.zip
-
Install the required packages:
pip install -r requirements.txt
-
Pre-process the dataset data using the
fatigueset_preprocess.pyscript we created:python fatigueset_preprocess.py --data_path=fatigueset --window_secs=5 --data_save_dir=processed --num_cores=4
3.1. Adjust the
--window_secs=5according to the time window you want to be used to accumulate and calculate statiscal data regarding the values in each time window from start to the end of the recorded data.3.2. Adjust the
--num_cores=4according to the amount of CPU cores you want to dedicate to the data preprocessing task. -
After running the script following the last intruction, there should be available a folder inside the
processed/folder containing data of a time window relative to the folder name, for the case above it should beprocessed/5s/(if ran without changes) for example. -
Don't be afraid to create an issue if you have suggestions to improve the data preprocessing process.
- [1] “FatigueSet: A Multi-modal Dataset for Modeling Mental Fatigue and Fatigability | Request PDF,” ResearchGate, Nov. 2024, doi: 10.1007/978-3-030-99194-4_14.
- [2] C. Kodikara, S. Wijekoon, and L. Meegahapola, “FatigueSense: Multi-Device and Multimodal Wearable Sensing for Detecting Mental Fatigue,” ACM Trans. Comput. Healthcare, vol. 6, no. 2, p. 14:1-14:36, Feb. 2025, doi: 10.1145/3709363.
- [3] J. Harrison, A. Sharma, C. Finn, and M. Pavone, “Continuous Meta-Learning without Tasks,” Oct. 21, 2020, arXiv: arXiv:1912.08866. doi: 10.48550/arXiv.1912.08866