Skip to content

akanametov/yolo-face

Repository files navigation


Ultralytics CI YOLOv8 Citation Docker Pulls
Run on Gradient Open In Colab Open In Kaggle

Ultralytics YOLOv8, developed by Ultralytics, is a cutting-edge, state-of-the-art (SOTA) model that builds upon the success of previous YOLO versions and introduces new features and improvements to further boost performance and flexibility. YOLOv8 is designed to be fast, accurate, and easy to use, making it an excellent choice for a wide range of object detection, image segmentation and image classification tasks.

Installation

# clone repo
git clone https://github.com/akanametov/yolov8-face

# pip install required packages
pip install ultralytics

# go to code folder
cd yolov8-face

Trained models

yolov8n-face.pt

yolov8m-football.pt

yolov8m-parking.pt

yolov8m-drone.pt

YOLOv8-face

Inference

On image:

yolo task=detect mode=predict model=yolov8n-face.pt conf=0.25 imgsz=1280 line_thickness=1 max_det=1000 source=examples/face.jpg

Results

PR curve:

Losses and mAP:

Confusion matrix:

Training

Data preparation

Single GPU training

# train model
yolo task=detect \
mode=train \
model=yolov8n.pt \
data=datasets/data.yaml \
epochs=100 \
imgsz=640

YOLOv8-football

Inference

On image:

yolo task=detect mode=predict model=yolov8m-football.pt conf=0.25 imgsz=1280 line_thickness=1 source=examples/football.jpg

Results

PR curve:

Losses and mAP:

Confusion matrix:

Training

Data preparation

Single GPU training

# train model
yolo task=detect \
mode=train \
model=yolov8m.pt \
data=datasets/data.yaml \
epochs=120 \
imgsz=960

YOLOv8-parking

Inference

On image:

yolo task=detect mode=predict model=yolov8m-parking.pt conf=0.25 imgsz=1280 line_thickness=1 source=examples/parking.jpg

Results

PR curve:

Losses and mAP:

Confusion matrix:

Training

Data preparation

Single GPU training

# train model
yolo task=detect \
mode=train \
model=yolov8m.pt \
data=datasets/data.yaml \
epochs=10 \
batch=32 \
imgsz=640

YOLOv8-drone

Inference

On image:

yolo task=detect mode=predict model=yolov8m-drone.pt conf=0.25 imgsz=1280 line_thickness=1 source=examples/drone.jpg

Results

PR curve:

Losses and mAP:

Confusion matrix:

Training

Data preparation

Single GPU training

# train model
yolo task=detect \
mode=train \
model=yolov8m.pt \
data=datasets/data.yaml \
epochs=100 \
imgsz=640

Transfer learning

yolov8n.pt

yolov8m.pt

License

YOLOv8 is available under two different licenses:

  • GPL-3.0 License: See LICENSE file for details.
  • Enterprise License: Provides greater flexibility for commercial product development without the open-source requirements of GPL-3.0. Typical use cases are embedding Ultralytics software and AI models in commercial products and applications. Request an Enterprise License at Ultralytics Licensing.

Contact

For YOLOv8 bugs and feature requests please visit GitHub Issues. For professional support please Contact Us.