Improving Human Activity Recognition Integrating LSTM with Different Data Sources: Features, Object Detection and Skeleton Tracking
Example of how generate features, 3D skeleton data and object detections and how to be trained with our integrated architecture
Before running the code:
- Download STAIR dataset and verify that all folders of the code point to the right location.
- Install OpenPose and YOLO v5.
- Generate YOLO, OpenPose and Features data with:
python3 paso1_procesarVideosYOLO.py assisting_in_getting_up end
python3 paso2_procesarVideosOpenPoseParalelo.py assisting_in_getting_up end
python3 paso3_getFeatures.py assisting_in_getting_up end
- Join data:
python3 paso4_generarDatosEntrenamiento_FeaturesOpenPoseYOLO_NPZ.py
- Train integrated model:
python3 paso5_entrenarAcciones_YOLO_OpenPose_Features.py
- Evaluate model:
python3 paso6_evaluarModeloF1Score.py
Optionally, for other datasets such as NTU-RGB-D, an initial distribution of data should be done:
python3 paso0_moverFicheros_NTU_RGB.py