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))