Face recognition 5.1- insightface face face detection model training practice notes

1、 insightface/ detection / scrfd model training

1.0、 Data sets

Data set download instructions ：https://github.com/deepinsight/insightface/tree/master/detection/datasets

Official download directory ：http://shuoyang1213.me/WIDERFACE/

Label Download ：https://gitee.com/link?target=https%3A%2F%2Fdrive.google.com%2Ffile%2Fd%2F1UW3KoApOhusyqSHX96yEDRYiNkd3Iv3Z%2Fview%3Fusp%3Dsharing

The propeller data set and label are packaged and downloaded as a whole ：https://aistudio.baidu.com/aistudio/datasetdetail/157445

1.1、 install

1.1.0、 Virtual environment installation

Linux-Centos7 Lower installation Anaconda,conda Installing a virtual environment ,cuda,cudnn

1.1.1、pytorch install

Different cuda Versions are installed in different ways , Find historical versions ：pytorch Install the reference

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

import torch
print(torch.__version__)  # '1.7.0'
print(torch.version.cuda)  # 10.2
torch.cuda.is_available() # False

1.1.2、mmcv install

mmcv Installation reference and cuda as well as pytorch Match ：https://mmcv.readthedocs.io/en/latest/get_started/installation.html

pip install -U openmim
mim install mmcv-full==1.3.17 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.7.0/index.html

pip install mmdet -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.7.0/index.html

1.1.3、 Overall compilation and installation

git clone https://gitee.com/AI-Mart/insightface.git

cd insightface/detection/scrfd
 perhaps 
cd /data/mart/face_insightface/insightface/detection/scrfd

pip install -r requirements/build.txt
pip install -v -e .  # or "python setup.py develop"
pip install scipy

1.2、 Training

python ./tools/train.py ./configs/scrfd/scrfd_1g.py --gpus=1

nohup python ./tools/train.py ./configs/scrfd/scrfd_1g.py --gpus=1 2>&1 &
tail -f nohup.out