Advanced pipelines for video datasets
Project description
Bumblebee
Bumblebee provides high level components to construct training pipelines for videos conveniently.
Install
pip install eyecu_bumblebee
Motivation
Everything should be made as simple as possible, but no simpler. - Albert Einstein
Our Websites
Examples
A pipeline with basic elements
from bumblebee import *
if __name__ == "__main__":
VIDEO_PATH = "/path/to/video.mp4"
# Create a source
file_stream = sources.FileStream(VIDEO_PATH)
# Add an effect
goto = effects.GoTo(file_stream)
END_OF_VIDEO = file_stream.get_duration()
goto(END_OF_VIDEO)
# Create a dataset
single_frame = datasets.SingleFrame(file_stream)
last_frame = single_frame.read()
Using Manager API
from bumblebee import *
if __name__ == "__main__":
# Create a training manager
manager = managers.BinaryClassification(
["path/to/video_dir","path/to/another_dir"],
["path/to/labels"]
)
number_of_epochs = 300
for epoch,(frame_no,frame,prob) in manager(number_of_epochs):
# Use data stuff
...
Read limited section of video
from bumblebee import *
from
if __name__ == "__main__":
VIDEO_PATH = "/path/to/video.mp4"
start_frame = 35
end_frame = 40
file_stream = sources.FileStream(VIDEO_PATH)
limited_stream = effects.Start(file_stream,start_frame)
limited_stream = effects.End(limited_stream,end_frame)
single_frame = datasets.SingleFrame(file_stream)
for frame in single_frame:
...
Team
This project is currently developed and maintained by ovuruska.
License
Bumblebee has MIT license. You can find further details in 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.
Source Distribution
eyecu-bumblebee-0.4.2.tar.gz
(12.9 kB
view hashes)
Built Distribution
Close
Hashes for eyecu_bumblebee-0.4.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e380228e4d3e6a3d745d34f7885a9a34330f93b5ad94532c88909ad192a1cf11 |
|
MD5 | 36c312bdc5594affe4f90dd517d3d351 |
|
BLAKE2b-256 | 6aacb6d9fa668bfec8d781252f76248a49a0eabfe60b6e6fdc3773d3daab1e4e |