In the YOLO9000 paper, they define the distance between a box and centroid as $d(box, centroid) = 1 - IOU(box, centroid)$. I think box here is a ground truth bounding box, but what is centroid?
$\begingroup$
$\endgroup$
Add a comment
|
1 Answer
$\begingroup$
$\endgroup$
The centroid is referring to the central point of a k-means cluster.
Where does k-means come in? At the end of p.2 they say:
Instead of choosing priors by hand, we run k-means clustering on the training set bounding boxes to automatically find good priors.
Also see Figure 2.
answered Jan 7, 2022 at 17:59