This repository provides a PyTorch implementation of the paper MS-TCN: Multi-Stage Temporal Convolutional Network for Action Segmentation.
An extended version has been published in TPAMI Link.
Tested with:
data
folder in the same directory as main.py
.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.Run python main.py --action=predict --dataset=DS --split=SP
.
Run python eval.py --dataset=DS --split=SP
.
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
S. Li, Y. Abu Farha, Y. Liu, MM. Cheng, and J. Gall.
MS-TCN++: Multi-Stage Temporal Convolutional Network for Action Segmentation.
In IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI), 2020