From 82b720b52d82f2d7cf1f5e9533c9470b6fb41b35 Mon Sep 17 00:00:00 2001 From: Ruben Barkow-Kuder Date: Wed, 7 Jul 2021 03:37:12 +0200 Subject: [PATCH] Readme: Add install instructions --- README.md | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index a3a8ab6..c7fd12e 100644 --- a/README.md +++ b/README.md @@ -20,6 +20,19 @@ _2[Department of Computing, The Hong Kong Polytechnic University](htt ## News (2021-07-06) The training code will be released soon. Stay tuned. +## Installation + +1. Clone this repository: +```bash +git clone https://github.com/yangxy/GPEN.git +cd GPEN +``` +2. install requirements: +``` +pip install -r requirements.txt +``` +*Note:* This will install about 6 GB of packages to your system. + ## Usage ![python](https://img.shields.io/badge/python-v3.7.4-green.svg?style=plastic) @@ -28,11 +41,7 @@ _2[Department of Computing, The Hong Kong Polytechnic University](htt ![driver](https://img.shields.io/badge/driver-v460.73.01-green.svg?style=plastic) ![gcc](https://img.shields.io/badge/gcc-v7.5.0-green.svg?style=plastic) -- Clone this repository: -```bash -git clone https://github.com/yangxy/GPEN.git -cd GPEN -``` + - Download RetinaFace model and our pre-trained model (not our best model due to commercial issues) and put them into ``weights/``. [RetinaFace-R50](https://public-vigen-video.oss-cn-shanghai.aliyuncs.com/robin/models/RetinaFace-R50.pth) | [GPEN-512](https://public-vigen-video.oss-cn-shanghai.aliyuncs.com/robin/models/GPEN-512.pth) | [GPEN-1024-Color](https://public-vigen-video.oss-cn-shanghai.aliyuncs.com/robin/models/GPEN-1024-Color.pth)