/YOLO-kmeans-Tool

Easy and useful yolo kmeans tool for anchor box

Primary LanguagePython

YOLO-kmeans-Tool

easy and useful yolo kmeans tool

Introduction

when we do detection work, set anchor box is the first thing you should think about. So this tool means to help you compute the anchor box of bboxes.

Usage

this tool is easy to use, you can run it like :

python kmeans.py 5 anofile.csv

the num '5' means the cluster that you want. the anofile.csv is ano bboxes file like : path,x1,y1,x2,y2,label

Reference