ArcFace/InsightFace使用自己数据训练/验证过程(3)

最新推荐文章于 2024-07-25 14:30:52 发布

Mr.小梅

最新推荐文章于 2024-07-25 14:30:52 发布

阅读量7.2k

点赞数 2

分类专栏：人脸识别 InsightFace训练过程文章标签： insightface自定义数据训练 arcface训练人脸识别训练过程

本文链接：https://blog.csdn.net/qq_39707285/article/details/98177506

版权

ArcFace/InsightFace使用自己数据训练/验证过程(3)

接上一章内容ArcFace/InsightFace使用自己数据训练/验证过程(2)
文章所需脚本文件下载地址

(六) 训练

进入src目录，linux命令行输入以下：

export MXNET_CPU_WORKER_NTHREADS=24
export MXNET_CUDNN_AUTOTUNE_DEFAULT=0
export MXNET_ENGINE_TYPE=ThreadedEnginePerDevice

DATA_DIR=/root/cc/faces_vgg_112x112	#数据存放目录
NETWORK=r50	#网络 默认resnet50 r100代表resnet100
JOB=softmax1e3	#
LOSS=0	#0代表使用softmax，4代表arcface等等，可以看源码
MODELDIR="/root/cc/model/model-$NETWORK-$JOB"	#模型保存位置
mkdir -p "$MODELDIR"
PREFIX="$MODELDIR/model"
CUDA_VISIBLE_DEVICES='0' python -u train_softmax.py --data-dir $DATA_DIR --network "$NETWORK" --loss-type $LOSS --prefix "$PREFIX" --per-batch-size 64