Cranfield-GDP3 / TACTUS-model

0 stars 0 forks source link

Rework processing pipeline #4

Closed MarcBresson closed 1 year ago

MarcBresson commented 1 year ago
 - initialise an empty array to store all the data

- For each video:
    - parse the labels.json file
    - parse the fps/yolov7.json file

    - for every data_augmentation:
        - for each frame:
             - initialise a dictionnary for every unique skeleton as they will require separate rolling windows (only if multiple people on the frame)
             - if the offender skeleton is tracked
                 - add it to the rolling window (or create the rolling window is the skeleton id didn't exist)
                 - process it (velocity, velocity centre, angles)
                 - store it

- save the file