【人工智能与机器学习】创新性大作业--人脸口罩数据集的模型训练

操作系统：Win10
编程环境：Jupyter Notebook
模型训练环境：Opencv-3.4.1
Python版本：python3.7

一、OpenCv的下载及安装

1、下载

opencv官网：https://opencv.org/opencv-3-4-1/

2、选择合适的版本进行下载安装

3、等待读秒，进行下载

4、选择合适的路径进行安装

二、下载人脸口罩数据集并进行处理

1、下载人脸口罩数据集

人脸口罩数据集链接如下：https://download.csdn.net/download/qq_42451251/12566250

2、数据集重命名为连续序列

将数据集重命名为连续序列，重命名代码如下：

#对数据集重命名
#coding:utf-8
import os
path = "F:\\mask\\have_mask" #你的路径
filelist = os.listdir(path)
count=1000 #开始文件名1000.jpg
for file in filelist:   
    Olddir=os.path.join(path,file)  
    if os.path.isdir(Olddir):  
        continue
    filename=os.path.splitext(file)[0]   
    filetype=os.path.splitext(file)[1]
 
    Newdir=os.path.join(path,str(count)+filetype)  
    os.rename(Olddir,Newdir)
    count+=1

正样本重命名结果如下所示：

同样的步骤对负样本进行重命名，而后在对正负样本数据集进行像素处理
代码如下：

#修改正样本像素
import pandas as pd
import cv2
for n in range(1000,1606):#代表正数据集中开始和结束照片的数字
    path='F:\\mask\\have_mask\\'<