logo
Browse Source

Update README

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 1 year ago
parent
commit
97af2e1619
  1. 35
      README.md
  2. BIN
      result.png

35
README.md

@ -19,35 +19,22 @@ and maps vectors with labels provided by the [Kinetics400 Dataset](https://deepm
Use the pretrained SLOWFAST model ('slowfast_r50')
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.pytorchvideo(model_name='slowfast_r50')
.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.pytorchvideo(model_name='slowfast_r50'))
.output('path', 'labels', 'scores', 'features')
)
```
<img src="./result1.png" width="800px"/>
*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.pytorchvideo['frames', ('labels', 'scores', 'features')](
model_name='slowfast_r50')
.select['path', 'labels', 'scores', 'features']()
.show(formatter={'path': 'video_path'})
)
DataCollection(p('./archery.mp4')).show()
```
<img src="./result2.png" width="800px"/>
<img src="./result.png" width="800px"/>
<br />

BIN
result.png

Binary file not shown.

After

Width:  |  Height:  |  Size: 15 KiB

Loading…
Cancel
Save