使用Tensorflow 检测是否正确佩戴口罩(二)
接上一篇文章,因为使用的数据集出现了数据泄露(测试集中的图像在训练集和验证集中出现),因此这次我们换一个数据集再来训练一次模型。
1、准备数据集
本次使用的数据集依然是在Kaggle上获得的,名称为Face Mask Detection。
这个数据集分成了两个部分,一个是图像文件,一个是标注文件。图像文件与上次不同,上次使用的是已经框好的人脸,这次用的图像就是生活中常见的图像,人脸只占其中的一小部分面积。
标注文件是一个.xml文件,我也是第一次接触这个文件类型,观察后发现,这个文件里面包含了图像的分类信息(‘with_mask’, ‘without_mask’, ‘mask_weared_incorrect’)、图像中人脸的位置 xmin、ymin、xmax、ymax,包含多张人脸得图像也给出了多个位置信息。我们要做得就是提取信息。
<annotation>
<folder>images</folder>
<filename>maksssksksss0.png</filename>
<size>
<width>512</width>
<height>366</height>
<depth>3</depth>
</size>
<segmented>0</segmented>
<object>
<name>without_mask</name>
<pose>Unspecified</pose>
<truncated>0</truncated>
<occluded>0</occluded>
<difficult>0</difficult>
<bndbox>
<xmin