deadpoppy / YOLO-kmeans-Tool

Easy and useful yolo kmeans tool for anchor box

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

Easy and useful yolo kmeans tool for anchor box


Languages

Language:Python 100.0%