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