Generating anchor boxes using K-means clustering , YOLO

Question

I am trying to understand working of YOLO and how it detects object in an image. My question is, what role does k-means clustering play in detecting bounding box around the object? Thanks.

Answer 1

K -means clustering algorithm is very famous algorithm in data science. This algorithm aims to partition n observation to k clusters. Mainly it includes :

1. Initialization : K means (i.e centroid) are generated at random.

2. Assignment : Clustering formation by associating the each observation with nearest centroid.

3. Updating Cluster : Centroid of a newly created cluster becomes mean.

Assignment and Update are repitatively occurs untill convergence. The final result is that the sum of squared errors is minimized between points and their respective centroids.

EDIT :

Why use K means

1. K-means is computationally faster and more efficient compare to other unsupervised learning algorithms. Don't forget time complexity is linear.
2. It produces a higher cluster then the hierarchical clustering. More number of cluster helps to get more accurate end result.
3. Instance can change cluster (move to another cluster) when centroid are re-computed.
4. Works well even if some of your assumption are broken.

what it really does in determining anchor box

1. It will create a thouasands of anchor box (i.e Clusters in k-means) for each predictor that represent shape, location, size etc.
2. For each anchor box, calculate which object’s bounding box has the highest overlap divided by non-overlap. This is called Intersection Over Union or IOU.
3. If the highest IOU is greater than 50% ( This can be customized), tell the anchor box that it should detect the object it has highest IOU.
4. Otherwise if the IOU is greater than 40%, tell the neural network that the true detection is ambiguous and not to learn from that example.
5. If the highest IOU is less than 40%, then it should predict that there is no object.

Thanks!