sample script for inference

Author: clibdev

.gitignore

@@ -1,4 +1,4 @@
-*/__pycache__
+**/*/__pycache__
 /.idea
 *.pt
 *.onnx

README.md

@@ -3,6 +3,7 @@
 Differences between original repository and fork:
 
 * Compatibility with PyTorch >=2.0. (🔥)
+* Sample script [detect.py](detect.py) for inference.
 
 # Installation
 
@@ -20,22 +21,8 @@ pip install -r requirements.txt
 | yolov8s         | 640        | 96.0  | 94.2   | 82.6  |  -             | -       | 
 | yolov8m         | 640        | 96.6  | 95.0   | 84.1  |  -             | -       |
 
+# Inference
 
-
-#### yolov8n-face
-
-![yolov8n-face](data/test.jpg)
-
-
-#### demo
-
-* [yolov8-face-landmarks-opencv-dnn](https://github.com/hpc203/yolov8-face-landmarks-opencv-dnn)
-
-
-#### References
-
-* [https://github.com/ultralytics/ultralytics](https://github.com/ultralytics/ultralytics)
-
-* [https://github.com/deepcam-cn/yolov5-face](https://github.com/deepcam-cn/yolov5-face)
-
-* [https://github.com/derronqi/yolov7-face](https://github.com/derronqi/yolov7-face)
+```shell
+python detect.py --weights yolov8n-face.pt --source ultralytics/assets/bus.jpg --save-img
+```

detect.py

@@ -0,0 +1,24 @@
+import argparse
+from ultralytics import YOLO
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--weights', type=str, default='./yolov8n-face.pt', help='model.pt path(s)')
+    parser.add_argument('--source', type=str, default='./ultralytics/assets/bus.jpg', help='source')
+    parser.add_argument('--img-size', nargs='+', type=int, default=640, help='inference size (pixels)')
+    parser.add_argument('--conf-thres', type=float, default=0.6, help='object confidence threshold')
+    parser.add_argument('--iou-thres', type=float, default=0.5, help='IOU threshold for NMS')
+    parser.add_argument('--device', default='0', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
+    parser.add_argument('--save-img', action='store_true', help='save results')
+    opt = parser.parse_args()
+
+    model = YOLO(opt.weights)
+    model.predict(
+        source=opt.source,
+        imgsz=opt.img_size,
+        conf=opt.conf_thres,
+        iou=opt.iou_thres,
+        device=opt.device,
+        save=opt.save_img
+    )

test_widerface.py

@@ -4,7 +4,7 @@
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument('--weights', nargs='+', type=str, default='runs/pose/yolov8n-face/weights/best.pt', help='model.pt path(s)')
+    parser.add_argument('--weights', nargs='+', type=str, default='./yolov8n-face.pt', help='model.pt path(s)')
     parser.add_argument('--img-size', nargs= '+', type=int, default=640, help='inference size (pixels)')
     parser.add_argument('--conf-thres', type=float, default=0.01, help='object confidence threshold')
     parser.add_argument('--iou-thres', type=float, default=0.5, help='IOU threshold for NMS')