Add --source screen Usage example (ultralytics#10215)

libaos · Nov 18, 2022 · 9bc6034 · 9bc6034
1 parent 9bf1855
commit 9bc6034
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/classify/predict.py b/classify/predict.py
@@ -6,6 +6,7 @@
     $ python classify/predict.py --weights yolov5s-cls.pt --source 0                               # webcam
                                                                    img.jpg                         # image
                                                                    vid.mp4                         # video
+                                                                   screen                          # screenshot
                                                                    path/                           # directory
                                                                    'path/*.jpg'                    # glob
                                                                    'https://youtu.be/Zgi9g1ksQHc'  # YouTube

diff --git a/detect.py b/detect.py
@@ -6,6 +6,7 @@
     $ python detect.py --weights yolov5s.pt --source 0                               # webcam
                                                      img.jpg                         # image
                                                      vid.mp4                         # video
+                                                     screen                          # screenshot
                                                      path/                           # directory
                                                      'path/*.jpg'                    # glob
                                                      'https://youtu.be/Zgi9g1ksQHc'  # YouTube

diff --git a/segment/predict.py b/segment/predict.py
@@ -6,6 +6,7 @@
     $ python segment/predict.py --weights yolov5s-seg.pt --source 0                               # webcam
                                                                   img.jpg                         # image
                                                                   vid.mp4                         # video
+                                                                  screen                          # screenshot
                                                                   path/                           # directory
                                                                   'path/*.jpg'                    # glob
                                                                   'https://youtu.be/Zgi9g1ksQHc'  # YouTube