本文主要介绍在前文训练结果的基础上进行测试。
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的使用。