File size: 681 Bytes
d670799 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
# Copyright (c) OpenMMLab. All rights reserved.
import platform
import decord
import pytest
from mmaction.structures import ActionDataSample
from mmaction.visualization import ActionVisualizer
@pytest.mark.skipif(platform.system() == 'Windows', reason='Windows mem limit')
def test_visualizer():
video = decord.VideoReader('./demo/demo.mp4')
video = video.get_batch(range(32)).asnumpy()
data_sample = ActionDataSample()
data_sample.set_gt_label(2)
vis = ActionVisualizer()
vis.add_datasample('demo', video)
vis.add_datasample('demo', video, data_sample)
vis.add_datasample('demo', video, data_sample, step=1)
return
|