Yolov4 和 yolov4-tiny 聚类k-means得到先验框

Yolov4 和 yolov4-tiny 聚类k-means得到先验框

之前查看大佬们的聚类办法,需要查看好多代码,现在源码里直接实现了,直接使用命令就行。
在训练模型:./darknet detector train data/train.data data/yolov4-tiny.cfg yolov4-tiny.conv.29 -map
之前,对于yolov4-tiny:使用命令:./darknet detector calc_anchors data/train.data -num_of_clusters 6 -width 416 -height 416
对于yolov4:使用命令:./darknet detector calc_anchors data/train.data -num_of_clusters 9 -width 416 -height 416
得到anchor 之后,修改你cfg 里对应的anchor数值。
能帮到你的就收藏点赞加关注吧。

你可能感兴趣的:(神经网络,机器学习,深度学习)