Pytorch下训练能识别口罩的libfacedetection

在Ubuntu16.04上安装好pytorch已经一段时间了，寻思着做点啥，不枉费前期的安装和GPU卡，接下来先拿于仕琪的libfacedetection的训练练练吧，于仕琪的libfacedetection对于人脸检测领域，是那么耀眼的一颗星，由于快、准、狠，大家争相使用。但也不是全部好用，如果长期使用，肯定要补充样本，比如，躺着带口罩的，坐在车里歪带着安全帽的，骑着电瓶车是否带头盔的，这些场景肯定需要自己加样本，进行训练。

git clone https://github.com/ShiqiYu/libfacedetection.train

其他的所需项目，按照README一步步处理即可。

import os
import shutil

def transform(src,dest):
    if not os.path.exists(src):  
        os.makedirs(src)  
    if os.path.exists(src):  
        for dir2 in os.listdir(src):  
           for fi in os.listdir(os.path.join(src,dir2)):  
               src_file = os.path.join(src,dir2,fi)
               des_file = os.path.join(dest,dir2+'_'+fi)  
               shutil.copy(src_file,des_file)
               print(des_file)

 
if __name__=='__main__':
    src_dir='/home/xxx/libfacedetection.train/data/WIDER_FACE_rect/images/'
    dest_dir='/home/xxx/libfacedetection.train/data/WIDER_FACE_rect/images1/'
    transform(src_dir,dest_dir)

于是，完成了数据的所有的处理，接下来，就可以训练了。

感觉到了pytorch的强大了吧，于仕琪的这个模型，比MTCNN快，比YOLO在人脸上准，与大多主流开源比较好很多，大家可以在理解和修改代码基础上，训练其他的模型。