此教程在ubuntu下执行,如果在windows下,需将./darknet换成你的darknet.exe的路径。
1. 训练模型
./darknet detector train [.data] [.cfg] [.weight]
说明:
[.data]是.data文件的路径。
[.cfg]是.cfg文件的路径。
[.weight]是预训练权重文件的路径,可以是.weight文件,也可以是.backup文件。
- 多GPU选择 :
./darknet detector train [.data] [.cfg] [.weight] -gpus 0, 1, 2
-gpus后面的数字是GPU的编号,用几块就写几个。
2. 测试模型
- 检测图片
./darknet detector test [.data] [.cfg] [.weight] [-thresh 0.25] [xxx.jpg]
- 检测视频
./darknet detector demo [.data] [.cfg] [.weight] [-thresh 0.25] [xxx.mp4]
- 批量检测
-dont_show -ext_output [.txt] result.txt
说明:
[.txt]为图片或视频文件的批量路径。
result.txt 为结果保存文件。
5. 调用网络摄像头
./darknet detector demo [.data] [.cfg] [.weight] [-thresh 0.25] [http://192.168.1.108:8080/video?dummy=x.mjpg -i 0]
- 调用手持端网络摄像头
首先需要下载mjpeg-stream软件
./darknet detector demo [.data] [.cfg] [.weight] [-thresh 0.25] [http://192.168.1.108:8080/video?dummy=IPWebcam -i 0]
或
./darknet detector demo [.data] [.cfg] [.weight] [-thresh 0.25] [http://192.168.1.108:8080/video?dummy=SmartWebcam -i 0]
3. 评价模型好坏
模型的好坏主要是看.weight权重文件
./darknet detector map [.data] [.cfg] [.weight]
参考:https://www.cnblogs.com/zhangAlin/p/10661768.html