update requirements.txt

Author: clibdev

.gitignore

@@ -0,0 +1,10 @@
+*/__pycache__
+/.idea
+*.pt
+*.onnx
+/data/widerface/val/images
+/widerface_evaluate/widerface_txt
+/widerface_evaluate/box_overlaps.c
+/widerface_evaluate/**/*.so
+/widerface_evaluate/**/*.o
+/runs

README.md

@@ -1,3 +1,15 @@
+# Fork of [derronqi/yolov8-face](https://github.com/derronqi/yolov8-face)
+
+Differences between original repository and fork:
+
+* Compatibility with PyTorch >=2.0. (🔥)
+
+# Installation
+
+```shell
+pip install -r requirements.txt
+```
+
 # yolov8-face
 
 | Method          |  Test Size | Easy  | Medium | Hard  | FLOPs (B) @640 | weights  |

requirements.txt

@@ -1,43 +1,10 @@
-# Ultralytics requirements
-# Usage: pip install -r requirements.txt
-
-# Base ----------------------------------------
-matplotlib>=3.2.2
-numpy>=1.21.6
-opencv-python>=4.6.0
-Pillow>=7.1.2
-PyYAML>=5.3.1
-requests>=2.23.0
-scipy>=1.4.1
-torch>=1.7.0
-torchvision>=0.8.1
-tqdm>=4.64.0
-
-# Logging -------------------------------------
-# tensorboard>=2.4.1
-# clearml
-# comet
-
-# Plotting ------------------------------------
-pandas>=1.1.4
-seaborn>=0.11.0
-
-# Export --------------------------------------
-# coremltools>=6.0  # CoreML export
-# onnx>=1.12.0  # ONNX export
-# onnxsim>=0.4.1  # ONNX simplifier
-# nvidia-pyindex  # TensorRT export
-# nvidia-tensorrt  # TensorRT export
-# scikit-learn==0.19.2  # CoreML quantization
-# tensorflow>=2.4.1  # TF exports (-cpu, -aarch64, -macos)
-# tflite-support
-# tensorflowjs>=3.9.0  # TF.js export
-# openvino-dev>=2022.3  # OpenVINO export
-
-# Extras --------------------------------------
-psutil  # system utilization
-thop>=0.1.1  # FLOPs computation
-# ipython  # interactive notebook
-# albumentations>=1.0.3
-# pycocotools>=2.0.6  # COCO mAP
-# roboflow
+torch>=2.0.0
+torchvision>=0.15.0
+opencv-python>=4.7.0
+tqdm>=4.65.0
+PyYAML>=6.0.0
+seaborn>=0.12.0
+scipy>=1.10.0
+thop>=0.1.1
+psutil>=5.9.0
+Cython>=0.29.35