From a20b0e158a162a2f9593c08e13c646fbdc6484e3 Mon Sep 17 00:00:00 2001 From: Th3NiKo Date: Thu, 6 Mar 2025 22:27:53 +0100 Subject: [PATCH] Fix parsing of arguments --- arguments_parser.py | 8 ++++---- video_heatmap.py | 4 ++++ 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/arguments_parser.py b/arguments_parser.py index 6831dbd..0472652 100644 --- a/arguments_parser.py +++ b/arguments_parser.py @@ -21,7 +21,7 @@ def prepare_parser(): heatmap and original video frames. Value between \ 0.0-1.0 represent what part of original video heatmap \ gonna take. Default: 0.9", - required=False, default=0.9) + required=False, default=0.9, type=float) # Disable live view parser.add_argument("-d", "--disable", action="store_true", dest="video_disable", @@ -38,11 +38,11 @@ def prepare_parser(): parser.add_argument("-s", "--skip", action="store", dest="video_skip", help="Skip first number of frames in order to warm up background \ substraction alghoritm. Default: 200 frames", - required=False, default=200) + required=False, default=200, type=int) - # Take every x frame + # Take every x frame parser.add_argument("-t", "--take-every", action="store", dest="take_every", help="In order to speed up process it is possible to skip frames and \ take every x frame. Default: 1 (take all frames).", - required=False, default=1) + required=False, default=1, type=int) return parser diff --git a/video_heatmap.py b/video_heatmap.py index 77d313c..f24783e 100644 --- a/video_heatmap.py +++ b/video_heatmap.py @@ -43,6 +43,10 @@ def main(): count = 0 + if args.take_every <= 0: + print("Take every argument should be positive and non-zero, setting it to default value 1") + args.take_every = 1 + while read_succes: read_succes, video_frame = capture.read() if read_succes: