基于yolov3的口罩检测

一、数据集准备 假设已经有了标注好的一分部数据, (1)运行voc2yolo3.py生成txt文件,保存在ImageSets/Main目录下 (2)修改voc_annotation.py中的classespython classes = ["mask", "unmask"] 运行后,生成三个txt文件,每一行对应的是一张图片的位置,标注的真实框位置以及label, 0表明戴口罩,1表明未戴口罩:
相关文章
相关标签/搜索