Module code >> skvideo.datasets
Fork me on GitHub

Source code for skvideo.datasets

import numpy as np

from os.path import dirname
from os.path import join

[docs]def bikes(): """Test video data for video scene detection algorithms Returns ------- path : string Absolute path to bikes.mp4 """ module_path = dirname(__file__) return join(module_path, 'data', 'bikes.mp4')
[docs]def bigbuckbunny(): """Test video data for general video algorithms Returns ------- path : string Absolute path to bigbuckbunny.mp4 """ module_path = dirname(__file__) return join(module_path, 'data', 'bigbuckbunny.mp4')
[docs]def fullreferencepair(): """Test video data for full-reference video quality algorithms The sequence "carphone" is provided for quality-based test cases where two videos of the same dimensions are needed. Returns ------- paths : ndarray, shape (2,) First element contains the absolute path to a pristine video. Second element contains absolute path to a distorted video. """ module_path = dirname(__file__) return np.array([ join(module_path, 'data', 'carphone_pristine.mp4'), join(module_path, 'data', 'carphone_distorted.mp4'), ])
__all__ = ['bikes', 'bigbuckbunny', 'fullreferencepair']