From 3d54044e8ccf5ba5db97cfb0fbead1a98523f21b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 25 Feb 2024 13:45:05 +0100 Subject: [PATCH] Add docstrings --- classify/predict.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/classify/predict.py b/classify/predict.py index b7d2f05d7bce..3139d82e7b7d 100644 --- a/classify/predict.py +++ b/classify/predict.py @@ -204,6 +204,7 @@ def run( def parse_opt(): + """Parses command line arguments for YOLOv5 inference settings including model, source, device, and image size.""" parser = argparse.ArgumentParser() parser.add_argument("--weights", nargs="+", type=str, default=ROOT / "yolov5s-cls.pt", help="model path(s)") parser.add_argument("--source", type=str, default=ROOT / "data/images", help="file/dir/URL/glob/screen/0(webcam)") @@ -229,6 +230,7 @@ def parse_opt(): def main(opt): + """Executes YOLOv5 model inference with options for ONNX DNN and video frame-rate stride adjustments.""" check_requirements(ROOT / "requirements.txt", exclude=("tensorboard", "thop")) run(**vars(opt))