Skip to content

Files

Latest commit

 

History

History
33 lines (22 loc) · 981 Bytes

action.md

File metadata and controls

33 lines (22 loc) · 981 Bytes

Finetune Action Classifier

Action classifier for powerlifting. Classify clip as squat, deadlift, or bench press.

Pretrain Backbone

The action classifier head uses DSTFormer as a backbone for getting motion represenation. The pretrain backbone can be trained from scratch. See Training Instruction. Or download pretrain backbone parameters here.

Preprocess Data

  1. Process Fit3D dataset. See Data Preprocessing

  2. Slice data into clips (len=243, stride=81)

python process_fit3d_action.py
  1. Verify data files in data/action/Fit3D

Train

Train the classifier using provided training script

python train_action.py \
--data_path /path/to/dataset \
--checkpoint /checkpoint/dir \
--pretrained /pretrain/model/dir \
--selection latest_epoch.bin \
--epochs 100 \
--batch_size 32