Add docstrings

ultralytics · Feb 25, 2024 · 3d54044 · 3d54044
1 parent a9a2716
commit 3d54044
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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))