Jetson nano 使用yolov3人脸戴口罩检测
前言
因为最近新型冠状病毒的影响,在网上无意间看见有人做出针对戴口罩检测的视频,当时觉得好厉害,为就想利用手中的jetson nano 也做一个。查找了很多资料,都没有找到,无意间看见 链接: 吉浦公司发的视频。根据他们的步骤我才得以成功。
一.下载Darknet
https://gitee.com/hechunmu/darknet.gitgit clone https://github.com/pjreddie/darknet.git
1.1 配置
cd darknet
sudo vim Makefile #修改Makefile
1.2 将Makefile的前三行修改一下
GPU=1
CUDNN=1
OPENCV=1
1.3 编译
make -j4
1.4 下载权重文件,这里直接下载tiny版的权重文件
wget https://pjreddie.com/media/files/yolov3-tiny.weights
下载darknet53.conv.74 && Yolov3-tiny.cfg
1.5. 测试
1.5.1照片的测试
./darknet detect cfg/yolov3-tiny.cfg yolov3-tiny.weights data/dog.jpg
1.5.2摄像头的测试
./darknet detector demo cfg/coco.data yolov3_416_416.cfg yolov3.weights -c 1
二.下载maskOD_nano
https://github.com/WhoseAI/maskOD_nano.git
2.1 新建文件夹yolo
2.1.1在yolo文件夹新建如图文件夹,然后把darknet复制到yolo文件夹里
2.1.2 把下载好的maskOD_nano里面的内容复制到/2_objDetect/mask 里。
2.1.3 新建 backup 文件夹
2.1.4 解压缩dataset.zip
2.1.5 运行“ python datadispatch.py”将帮助您完成3个任务
将所有.xml格式转换为.txt格式并存储在同一文件夹中
2.1.6 修改“ mask.data”,为每个提供正确的路径
2.1.7 修改“ yolov3-tiny.cfg”
classes = 2 @ Line135 + 177,原始值应为80
max_batches = 5000 @ Line20,原始值应为500200
filter = 21 @ Line127 + 171,原始值应为255
出于培训目的,修改“ yolov3-tiny.cfg”
[net]
# Testing
#batch=1