Retinaface inference
0. 本机
Ubuntu 18.04
gcc version 7.5.0
CUDA Driver Version / Runtime Version 11.2 / 10.2
Nvidia Driver Versioin 460.32.03
Anaconda3
1. 环境配置
- 安装 python 3.6
- 安装 pytorch 1.7.1
$ conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
- 安装 pip
$ sudo apt install python3-pip
- 安装 pycocotools
$ pip install pycocotools
- 安装 opencv
$ conda install opencv
2. 使用Retinaface进行预测
- 模型路径: