遮罩检测
该Python脚本用于检测给定图像中的人是否戴着口罩。 使用Darknet训练了自定义数据集。
实时检测的用法:
python yolo-live-cv2.py --yolo yolo
从图像检测的用法:
python custom_detector.py --image“ / path / to / image” --config“ /cfg/yolov3-custom.cfg” --weights“ weights / yolov3-custom_final.weights” --names“ class_names / voc.names”
杂项文件
data_augmentation.py是一个Python脚本,用于增加数据集使用数据扩充的范围。 named.py是根据正确的命名规则将图像和带注释的.txt文件重命名为相同的名称。 test_train_split.py用于生成测试并训练.txt文件