利用 DarkNet 训练和测试神经网络的常用命令

674 阅读1分钟

此教程在ubuntu系统下执行,如果在windows下,需将./darknet换成你的darknet.exe的路径。

一. 训练模型

./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的编号,用几块就写几个。

二. 测试模型

  1. 检测图片
./darknet detector test [.data] [.cfg] [.weight] [-thresh 0.25] [xxx.jpg]
  1. 检测视频
./darknet detector demo [.data] [.cfg] [.weight] [-thresh 0.25] [xxx.mp4]
  1. 批量检测
-dont_show -ext_output [.txt] result.txt

说明:
[.txt]为图片或视频文件的批量路径。
result.txt 为结果保存文件。

4. 调用网络摄像头

./darknet detector demo [.data] [.cfg] [.weight] [-thresh 0.25] [http://192.168.1.108:8080/video?dummy=x.mjpg -i 0]

5. 调用手持端网络摄像头 首先需要下载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]

三. 评价模型好坏

模型的好坏主要是看.weight权重文件

./darknet detector map [.data] [.cfg] [.weight]

参考:www.cnblogs.com/zhangAlin/p…