A simple and efficient wrapper for reading videos as NumPy tensors
Project description
Mydia
Reading videos into NumPy arrays was never more simple. This library provides an entire range of additional functionalities such as custom frame selection, frame resizing, pixel normalization, grayscale conversion and much more.
Getting started
1. Read a video, given its path
# Import
from mydia import Videos
# Initialize video path
video_path = r".docs/examples/sample_video/bigbuckbunny.mp4"
# Create a reader object
reader = Videos()
# Call the 'read()' function to get the video tensor
# which will be of shape (1, 132, 720, 1280, 3)
video = reader.read(video_path)
The tensor can be interpreted as:
- 1 video
- Having 132 frames,
- Dimension (width x height) of each frame: 1280x720 pixels
3
denotes that the video is in RGB format
2. Use multiple workers for reading the videos in parallel
from mydia import Videos
video_paths = [
"path/to/video_1",
"path/to/video_2",
"path/to/video_3",
...,
]
reader = Videos()
video = reader.read(video_path, workers=4)
View detailed examples here
Installation
-
Install using Conda Package Manager (Recommended):
conda install -c mrinaljain17 mydia
-
Install from PyPI:
pip install mydia
-
Alternatively, install from source:
First, clone the repository
git clone https://github.com/MrinalJain17/mydia.git
Then, build the module
cd mydia python setup.py install
Requirements
Python 3.x
(preferably from the Anaconda Distribution)
The program uses ffmpeg-python, which provides python bindings for FFmpeg (used as the backend for reading and processing videos).
To install FFmpeg
on your machine:
For Linux users -
$ sudo apt-get update
$ sudo apt-get install libav-tools
OR
$ sudo apt-get update
$ sudo apt-get install ffmpeg
For Windows or MAC/OSX users -
Download the required binaries from here.
Extract the zip file and add the location of binaries to the PATH
variable.
Required Libraries
Install the following packages along with their dependencies:
- ffmpeg-python
- Numpy
- tqdm - Required for displaying the progress bar.
pip install ffmpeg-python numpy tqdm
License
Copyright 2018 Mrinal Jain.
Released under the MIT License.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.