Skip to content
No description, website, or topics provided.
Branch: master
Clone or download
Latest commit 240a674 Mar 6, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
LICENSE Create LICENSE Mar 5, 2019
README.md Update README.md Mar 6, 2019
batch_gen.py add files Feb 26, 2019
eval.py
main.py add files Feb 26, 2019
model.py add files Feb 26, 2019

README.md

MS-TCN: Multi-Stage Temporal Convolutional Network for Action Segmentation

This repository provides a PyTorch implementation of the paper MS-TCN: Multi-Stage Temporal Convolutional Network for Action Segmentation.

Tested with:

  • PyTorch 0.4.1
  • Python 2.7.12

Training:

  • Download the data folder, which contains the features and the ground truth labels. (~30GB)
  • Extract it so that you have the data folder in the same directory as main.py.
  • To train the model run python main.py --action=train --dataset=DS --split=SP where DS is breakfast, 50salads or gtea, and SP is the split number (1-5) for 50salads and (1-4) for the other datasets.

Prediction:

Run python main.py --action=predict --dataset=DS --split=SP.

Evaluation:

Run python eval.py --dataset=DS --split=SP.

Citation:

If you use the code, please cite

Y. Abu Farha and J. Gall.
MS-TCN: Multi-Stage Temporal Convolutional Network for Action Segmentation.
In IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2019
You can’t perform that action at this time.