scrfd环境配置

scrfd-github
1.激活环境

conda create --name openmmlab python=3.8 -y
conda activate openmmlab

2.下载torch

pip install torch==1.10.0+cu113 torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html

3.下载mmcv

pip install mmcv-full==1.4.0 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html

4.原文是去pip install -r requirements/build.txt,但是这样安装的是最新版的cython，有点问题,我选择

pip install cython==0.29.28

5.安装mmdet

python setup.py develop

6.安装onnx

pip install onnx
pip install onnx-simplifier

7.一些其他的库

conda install matplotlib
conda install scipy

8.demo运行看看

python demo/image_demo.py weights/face.jpg configs/scrfd/scrfd_500m.py weights/model_500M.pth

效果

9.onnx导出

python tools/scrfd2onnx.py configs/scrfd/scrfd_34g.py weights/scrfd_34g.pth --shape 640 640 --input-img weights/face.jpg

显示mmcv版本不对，换成1.3.17

pip install mmcv-full==1.3.17 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html