该数据集包含多种条件下驾驶员疲劳状态的视频片段和相关信息,旨在用于开发与评估监测驾驶员疲劳程度的技术模型。
本数据集包含了人疲劳时的一些照片,建议训练时可以把打哈欠张嘴的状态和闭眼的状态作为疲劳标准,以此来进行一个新手练习的小项目。经过测试发现,由于原数据集中存在图片数据与标注数据不匹配的问题,故我们需要将这部分不匹配的数据删除。
以下是参考代码:
```python
import os, shutil
jpeg_path = Dataset/dataset/JPEGImages
jpeg_list = os.listdir(jpeg_path)
anno_path = Dataset/dataset/Annotations
anno_list = os.listdir(anno_path)
for pic in jpeg_list:
name = pic.split(.)[0]
anno_name = name + .xml
if anno_name not in anno_list:
os.remove(os.path.join(jpeg_path, pic))
```