copied
Readme
Files and versions
Updated 2 years ago
action-classification
Action Classification with TimeSformer
Author: Jael Gu
Description
An action classification operator generates labels of human activities (with corresponding scores) and extracts features for the input video. It transforms the video into frames and loads pre-trained models by model names. This operator has implemented pre-trained models from TimeSformer and maps vectors with labels.
Code Example
Use the pretrained TimeSformer model ('timesformer_k400_8x224') to classify and generate a vector for the given video path './archery.mp4' (download).
Write the pipeline in simplified style:
import towhee
(
towhee.glob('./archery.mp4')
.video_decode.ffmpeg()
.action_classification.timesformer(model_name='timesformer_k400_8x224')
.show()
)
Write a same pipeline with explicit inputs/outputs name specifications:
import towhee
(
towhee.glob['path']('./archery.mp4')
.video_decode.ffmpeg['path', 'frames']()
.action_classification.timesformer['frames', ('labels', 'scores', 'features')](
model_name='timesformer_k400_8x224')
.select['path', 'labels', 'scores', 'features']()
.show(formatter={'path': 'video_path'})
)
Factory Constructor
Create the operator via the following factory method
action_classification.timesformer( model_name='timesformer_k400_8x224', skip_preprocess=False, classmap=None, topk=5)
Parameters:
model_name: str
The name of pre-trained model. Supported model names:
- timesformer_k400_8x224
skip_preprocess: bool
Flag to control whether to skip UniformTemporalSubsample in video transforms, defaults to False. If set to True, the step of UniformTemporalSubsample will be skipped. In this case, the user should guarantee that all the input video frames are already reprocessed properly, and thus can be fed to model directly.
classmap: Dict[str: int]:
Dictionary that maps class names to one hot vectors. If not given, the operator will load the default class map dictionary.
topk: int
The topk labels & scores to present in result. The default value is 5.
Interface
A video classification operator generates a list of class labels and a corresponding vector in numpy.ndarray given a video input data.
Parameters:
video: List[towhee.types.VideoFrame]
Input video data should be a list of towhee.types.VideoFrame representing video frames in order.
Returns:
labels, scores, features: Tuple(List[str], List[float], numpy.ndarray)
- labels: predicted class names.
- scores: possibility scores ranking from high to low corresponding to predicted labels.
- features: a video embedding in shape of (768,) representing features extracted by model.
Jael Gu
7c52483f25
| 8 Commits | ||
---|---|---|---|
.gitattributes |
1.1 KiB
|
2 years ago | |
README.md |
3.0 KiB
|
2 years ago | |
__init__.py |
692 B
|
2 years ago | |
kinetics_400.json |
10 KiB
|
2 years ago | |
requirements.txt |
63 B
|
2 years ago | |
result1.png |
14 KiB
|
2 years ago | |
result2.png |
93 KiB
|
2 years ago | |
timesformer.py |
3.4 KiB
|
2 years ago |