diff --git a/README.md b/README.md index 69f3518..ae2e61f 100644 --- a/README.md +++ b/README.md @@ -19,37 +19,22 @@ and maps vectors with labels. Use the pretrained TimeSformer model ('timesformer_k400_8x224') to classify and generate a vector for the given video path './archery.mp4' ([download](https://dl.fbaipublicfiles.com/pytorchvideo/projects/archery.mp4)). - *Write the pipeline in simplified style*: +*Write a pipeline with explicit inputs/outputs name specifications:* ```python -import towhee - -( - towhee.glob('./archery.mp4') - .video_decode.ffmpeg() - .action_classification.timesformer(model_name='timesformer_k400_8x224') - .show() +from towhee.dc2 import pipe, ops, DataCollection + +p = ( + pipe.input('path') + .map('path', 'frames', ops.video_decode.ffmpeg()) + .map('frames', ('labels', 'scores', 'features'), + ops.action_classification.timesformer(model_name='timesformer_k400_8x224')) + .output('path', 'labels', 'scores', 'features') ) -``` - - -
-*Write a same pipeline with explicit inputs/outputs name specifications:* - -```python -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'}) -) +DataCollection(p('./archery.mp4')).show() ``` - +
diff --git a/requirements.txt b/requirements.txt index a0a47cd..53e3d75 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ -towhee.models>=0.7.0 - -towhee>=0.7.0 +# towhee.models>=0.7.0 +# towhee>=0.7.0 +einops scipy torchvision torch diff --git a/result.png b/result.png new file mode 100644 index 0000000..fdcaa1e Binary files /dev/null and b/result.png differ diff --git a/result1.png b/result1.png deleted file mode 100644 index 66c4758..0000000 Binary files a/result1.png and /dev/null differ diff --git a/result2.png b/result2.png deleted file mode 100644 index 829d819..0000000 Binary files a/result2.png and /dev/null differ