diff --git a/models/yolo.py b/models/yolo.py index 06b80032d3d3..2844cd0410e0 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -1,4 +1,8 @@ -# YOLOv5 YOLO-specific modules +"""YOLOv5-specific modules + +Usage: + $ python path/to/models/yolo.py --cfg yolov5s.yaml +""" import argparse import logging