diff --git a/README.md b/README.md index e42f50d..1cbb104 100644 --- a/README.md +++ b/README.md @@ -8,7 +8,7 @@ -## Desription +## Description This operator detects faces in the images by using [RetinaFace](https://arxiv.org/abs/1905.00641) Detector. It will return the bounding box positions and the confidence scores of detected faces. This repo is an adaptaion from [biubug6/Pytorch_Retinaface](https://github.com/biubug6/Pytorch_Retinaface). diff --git a/requirements.txt b/requirements.txt index 08ed5ee..881ac86 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,3 @@ -torch \ No newline at end of file +torch>=1.9.0 +torchvision>=0.10.0 +