From d58e2a74e5728daeed85790120974baf80572f09 Mon Sep 17 00:00:00 2001 From: Zacchaeus Scheffer Date: Tue, 21 Mar 2023 15:07:28 -0700 Subject: [PATCH] Add ability to change main() opts as input --- yolov5/detect.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/yolov5/detect.py b/yolov5/detect.py index 153f2c2..ed1e6b4 100644 --- a/yolov5/detect.py +++ b/yolov5/detect.py @@ -260,8 +260,9 @@ def parse_opt(): return opt -def main(): - opt = parse_opt() +def main(opt=None): + if opt is None: + opt = parse_opt() check_requirements(exclude=('tensorboard', 'thop')) run(**vars(opt))