我用的是pytorch版本(Windows 10系统,python 3.7),其项目地址在:https://github.com/TreB1eN/InsightFace_Pytorch

预训练模型IR-SE50和Mobilefacenet的结果地址:https://pan.baidu.com/s/12BUjjwy1uUTEF9HCx5qvoQ    和   https://pan.baidu.com/s/1hqNNkcAjQOSxUjofboN6qg

训练集emore的地址:https://pan.baidu.com/s/1eXohwNBHbbKXh5KHyItVhQ

准备工作结束,开始训练自己的模型:

1、下载mtcnn-pytorch模型(注意:是在InsightFace_Pytorch文件夹下):

git clone https://github.com/TropComplique/mtcnn-pytorch.git

这里不建议用这个指令,因为项目中本身包含了mtcnn_pytorch文件,git指令的这个文件夹里缺少部分文件,所以这步可以忽略。

2、数据准备:

(1)准备Facebank(用于通过摄像机或视频进行测试)

在data / face_bank文件夹中提供要检测的面部图像,也就是自己的应用数据,并确保其结构如下:

data/facebank/
        ---> id1/
            ---> id1_1.jpg
        ---> id2/
            ---> id2_1.jpg
        ---> id3/
            ---> id3_1.jpg
           ---> id3_2.jpg

(2)将预训练的模型下载到work_space / model

(3)准备训练集用于训练

将emore数据集解压到data数据集下,运行:

python prepare_data.py

此时遇到问题:

1)mxnet的安装:

pip install --user mxnet

得到结果如下(时间会有点长,毕竟有5822653张图):

faces_emore/
            ---> agedb_30
            ---> calfw
            ---> cfp_ff
            --->  cfp_fp
            ---> cfp_fp
            ---> cplfw
            --->imgs
            ---> lfw
            ---> vgg2_fp

(4)通过视频检测:

python infer_on_video.py -f [video file name] -s [save file name]

视频文件应位于data / face_bank文件夹中。

1)报错:ModuleNotFoundError: No module named 'tensorboardX'

      解决:

pip install tensorboardX

2)报错:ValueError: Object arrays cannot be loaded when allow_pickle=False

     解决:降numpy的版本:

pip install --user numpy==1.16.2

3)报错:ModuleNotFoundError: No module named 'mtcnn_pytorch',这个问题就很迷。

      解决:

pip install --user mtcnn_pytorch

 (5)摄像头检测:

1)调用摄像头拍照的指令:

python take_pic.py -n name

t拍照,q退出,如果出现两个人就优先识别可能性最高的那个。

2)将原来存在的照片放在facebank里面,文件结构如下:

- facebank/
         name1/
             photo1.jpg
             photo2.jpg
             ...
         name2/
             photo1.jpg
             photo2.jpg
             ...
         .....

然后运行:

python face_verify.py 

1)会出现一个错误:No such file or directory: 'work_space\\save\\model_cpu_final.pth'

解决方案:放入model中的两个.pth文件将一个更该名字为model_cpu_final,pth,然后再work_space文件夹中新建save文件夹,将.pth文件放入文件夹中。

2)另外,问题:RuntimeError: Error(s) in loading state_dict for Backbone:

解决:

(6)训练:

python train.py -b [batch_size] -lr [learning rate] -e [epochs]

# python train.py -net mobilefacenet -b 200 -w 4

 

posted on 2020-12-08 15:02  Fiona_Y  阅读(1764)  评论(2编辑  收藏  举报