From 8e5f67d4086c5cdd05f1a455f5aea330bb00f34e Mon Sep 17 00:00:00 2001 From: wxywb Date: Mon, 11 Apr 2022 16:30:06 +0800 Subject: [PATCH] update the torchvision requirement. Signed-off-by: wxywb --- README.md | 2 +- requirements.txt | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) 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 +