操作系统: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\\'<