Script for heatmap generation (for movement) from single video with static camera.
Program pipeline:
- Load frame
- Substract background
- Accumulate movement
- Normalize accumulated frame (for view only)
- Apply heatmap colormap
- Optional: Supervise with original frame (check --alpha parameter)
- Save new frame
Program structure:
- video_heatmap.py - main program with whole pipeline implemented
- vision.py - image operations module
- arguments_parser.py - manage script arguments
Tested with python 3.8.3.
Libraries used: opencv-python. (Tested on ver. 4.4.0)
You can install it using pip.
pip install -r requirements.txtCheck if script works and show help
python video_heatmap.py -hvideo_heatmap.py [-h] -f VIDEO_FILE [-a VIDEO_ALPHA] [-d] [-o VIDEO_OUTPUT] [-s VIDEO_SKIP] [-t TAKE_EVERY]
optional arguments:
-h, --help show this help message and exit
-f VIDEO_FILE, --file VIDEO_FILE
Video file path for which heatman will be created. Example: input.mp4
-a VIDEO_ALPHA, --alpha VIDEO_ALPHA
Optional parameter to adjust alpha between heatmap and original video frames.
Value between 0.0-1.0 represent what part of original video heatmap gonna take.
Default: 0.9
-d, --disable Disable live view of heatmap generation.
-o VIDEO_OUTPUT, --output VIDEO_OUTPUT
Adjust name of output files. Script creates two files one video .mp4 and one image .png.
Default: output
-s VIDEO_SKIP, --skip VIDEO_SKIP
Skip first number of frames in order to warm up background substraction alghoritm.
Default: 200 frames
-t TAKE_EVERY, --take-every TAKE_EVERY
In order to speed up process it is possible to skip frames and take every x frame.
Default: 1 (take all frames).Example usage:
python video_heatmap.py -f input.mp4 -o output_nameIf u want to DISABLE LIVE VIEW of process use -d flag
python video_heatmap.py -f input.mp4 -o output_name -dIf u want to SPEED UP PROCESS you can adjust -t flag to take every x frame.
Example: Take every third frame
python video_heatmap.py -f input.mp4 -o output_name -t 3It is good idea to warm up background substraction alghoritm and skip first x frames.
If u want to do it ADJUST NUMBER OF WARMUP FRAMES use -s flag
python video_heatmap.py -f input.mp4 -o output_name -s 100If u want to CHANGE ALPHA BETWEEN ORIGINAL AND HEATMAP use -a flag
python video_heatmap.py -f input.mp4 -o output_name -a 0.5Original video:
Output of script:
React
Typescript
Django
Laravel
Facebook
Microsoft
Google
Alibaba
D3
Tencent