Skip to content

Commit

Permalink
Update chapters/en/unit2/cnns/yolo.mdx
Browse files Browse the repository at this point in the history
Co-authored-by: A Taylor <112668339+ATaylorAerospace@users.noreply.github.com>
  • Loading branch information
sezan92 and ATaylorAerospace authored Oct 1, 2024
1 parent cf087f3 commit c64f46b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion chapters/en/unit2/cnns/yolo.mdx
Original file line number Diff line number Diff line change
Expand Up @@ -203,7 +203,7 @@ $$
Inference is simple. We pass the image through the network and get the $S \times S \times (B \times 5 + C)$ feature map. We then filter out the boxes which have confidence scores less than a threshold.

##### Non-Maximum Suppression
In rare cases, for large objects, the network tends to predict multiple boxes from multiple grid cells. To eliminate duplicate detections, we use a technique called non-maximum suppression (NMS). NMS works by selecting the box with the highest confidence score and eliminating all other boxes with an IOU greater than a threshold. This is done iteratively until no overlapping boxes remain.
In rare cases, for large objects, the network tends to predict multiple boxes from multiple grid cells. To eliminate duplicate detections, we use a technique called Non-Maximum Suppression (NMS). NMS works by selecting the box with the highest confidence score and eliminating all other boxes with an IOU greater than a threshold. This is done iteratively until no overlapping boxes remain.


The end to end flow looks like this:
Expand Down

0 comments on commit c64f46b

Please sign in to comment.