diff --git a/.gitignore b/.gitignore
index b4087c2..d9b1421 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,6 +3,7 @@ venv
weights
runs
*.pt
+*.onnx
WIDER_train.zip
WIDER_val.zip
wider_face_split.zip
diff --git a/README.md b/README.md
index d51b822..b693178 100644
--- a/README.md
+++ b/README.md
@@ -5,6 +5,7 @@ been pre-trained by Lindevs from scratch.
## Release Notes
+* **[2024-11-01]** Re-saved and re-uploaded PyTorch models to avoid the dill package usage warning.
* **[2023-12-09]** YOLOv8x-Face model has been added.
* **[2023-12-02]** YOLOv8n-Face, YOLOv8s-Face, YOLOv8m-Face and YOLOv8l-Face models has been added.
@@ -25,11 +26,11 @@ The models have been trained on [WIDERFace](http://shuoyang1213.me/WIDERFACE/) d
| Name | Model Size (MB) | Link | SHA-256 |
|--------------|-----------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------|
-| YOLOv8n-Face | 6.3
12.2 | [PyTorch](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8n-face-lindevs.pt)
[ONNX](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8n-face-lindevs.onnx) | a72e65818b9d61b15d6b5f58996b2391c827f4508c5725706cf61d05a9a70e49
3ebbeaddfe4fc51a9ab119a4c67cc6193b559b07a8104d029ba0e146fcaa2ad4 |
-| YOLOv8s-Face | 22.5
44.7 | [PyTorch](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8s-face-lindevs.pt)
[ONNX](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8s-face-lindevs.onnx) | 3c9962b02c911b0ff6e4bb5d634f6e72d0d8ba24ca3442287f280ba50bfe2f73
0bf65e2576c05f25a5d5454b57b313d54d41495b044e94eae25bb1205e5d8d18 |
-| YOLOv8m-Face | 52.0
103.6 | [PyTorch](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8m-face-lindevs.pt)
[ONNX](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8m-face-lindevs.onnx) | 2d96e2eac09fcac4a677664680beee1c210041d0eb7e2f6a434fb806d455b2dc
5c40fbed7e8c2328ccaca718eb9ce49e5d631ae622c275dbd569ef4feab70ebd |
-| YOLOv8l-Face | 87.6
174.7 | [PyTorch](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8l-face-lindevs.pt)
[ONNX](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8l-face-lindevs.onnx) | a24036da36c2b8ce5e2985f70b3f8bc0bd1df3941f48e91ec84b48bdd73345b4
5037c8362da630935b35f32670b732d4d196bfcd6eff5052704bb9a568955e7f |
-| YOLOv8x-Face | 136.7
272.7 | [PyTorch](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8x-face-lindevs.pt)
[ONNX](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8x-face-lindevs.onnx) | 1985c9b08ee443f0fff646cc0371dbff9825415fb5e33d33bd2dc14bd73a56d7
28d9e9a85e8e739f824f280b0f426cc77b0984ebac6dbc30817ae8e3336ac97e |
+| YOLOv8n-Face | 6.0
11.7 | [PyTorch](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8n-face-lindevs.pt)
[ONNX](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8n-face-lindevs.onnx) | b038ca653b503453a94f6e12d76feca6840b2a97d7a1322b4498c5e922f29832
8d0bfb0c3383c5bd7a78dd24ef79a21e2aa456619b6ab5e53867092d1c7dc414 |
+| YOLOv8s-Face | 21.5
42.7 | [PyTorch](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8s-face-lindevs.pt)
[ONNX](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8s-face-lindevs.onnx) | fa7e47fe9378255e4b52cb7abc4e387c0353dd26b0b8e6834045dc9dfbaaf69f
0a6d19f2f68d7f0cc8104ab5c9eaa54b63e298f91dcfefd4be897f94a1561d02 |
+| YOLOv8m-Face | 49.6
98.8 | [PyTorch](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8m-face-lindevs.pt)
[ONNX](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8m-face-lindevs.onnx) | 303dcd997fb6ed446d1626b2bbd36f146894cdf600e33c4d563124f8c1b191c4
652f1ee6cd0291295de3d8fcaf9375ad62ef269055c0ada458bfdc4e7e6095da |
+| YOLOv8l-Face | 83.6
166.6 | [PyTorch](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8l-face-lindevs.pt)
[ONNX](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8l-face-lindevs.onnx) | 29cc43b27c8b865859c66489a4399a10a3efd80ce68ded9815364117641706d5
52dc39e46a7316398c95d30dd669a641382c9fdd8b675ad32aa65585bf820ea0 |
+| YOLOv8x-Face | 130.4
260.1 | [PyTorch](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8x-face-lindevs.pt)
[ONNX](https://github.com/lindevs/yolov8-face/releases/latest/download/yolov8x-face-lindevs.onnx) | 117c587c79e75e68a83e70200549bf6c035fa45b30e02fb41699639aadcfa0e6
0ddad01728bc5f7d6c68c9b5567cfd1c8257f041af607c215ed865c5442f87fa |
* Training results:
diff --git a/requirements.txt b/requirements.txt
index 69a3592..b4e8ffe 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
-ultralytics>=2.1.0
+ultralytics>=8.3.0
### Optional - ONNX format
-# onnx>=1.15.0
+# onnx>=1.17.0
### Optional - WIDERFace evaluation
# Cython>=3.0.0
diff --git a/widerface/setup.py b/widerface/setup.py
index 47716fb..767205e 100644
--- a/widerface/setup.py
+++ b/widerface/setup.py
@@ -1,4 +1,4 @@
-from distutils.core import setup, Extension
+from setuptools import setup, Extension
from Cython.Build import cythonize
import numpy