基於yolov3的口罩檢測

一、數據集準備 假設已經有了標註好的一分部數據, (1)運行voc2yolo3.py生成txt文件,保存在ImageSets/Main目錄下 (2)修改voc_annotation.py中的classespython classes = ["mask", "unmask"] 運行後,生成三個txt文件,每一行對應的是一張圖片的位置,標註的真實框位置以及label, 0表明戴口罩,1表明未戴口罩:
相關文章
相關標籤/搜索