YoloV4训练自己的数据集(五)

本文主要介绍在前文训练结果的基础上进行测试。

1.修改配置文件

本次需要把上文的yolov4-custom.cfg再复制一份，命名为yolov4-custom-test.cfg。之后修改第6行，第7行参数为：

batch=1
subdivisions=1

2.测试图片

使用以下命令测试图片：

./darknet detector test cfg/voc.data cfg/yolov4-custom-test.cfg backup/yolov4-custom-train.weights testfiles/2

其中testfiles是需要测试图片的文件夹，2是图片的名字（建议图片命名的时候不要带.jgp的后缀，带了可能会报错）。

3.测试视频

./darknet detector demo cfg/voc.data cfg/yolov4-custom-test.cfg backup/yolov4-custom-train.weights testfiles/1.mp4

其中 1.mp4是视频的名字。
至此，训练以及测试结束。
下文将介绍Yolo-tiny的使用。