Non-max ์–ต์ œ

  • 19x19 ์˜ ๊ฒฉ์ž๋กœ YOLO ๋ฅผ ์‹คํ–‰ํ–ˆ์„ ๋•Œ ์—ฌ๋Ÿฌ ์…€์—์„œ ์ž๋™์ฐจ๋ฅผ ์ธ์‹ํ•  ์ˆ˜ ์žˆ๋‹ค.

  • ํ•œ ์…€์—์„œ๋งŒ ์ž๋™์ฐจ๊ฐ€ ์žˆ๋‹ค๊ณ  ์ธ์‹ํ•˜๋„๋ก ํ•˜๋ ค๋ฉด ์–ด๋–ป๊ฒŒ ํ•ด์•ผํ• ๊นŒ?

  • ์ด๋ฅผ ์œ„ํ•ด์„œ ๊ฐ™์€ ๊ฐ์ฒด๋ฅผ ์ธ์‹ํ•œ ์—ฌ๋Ÿฌ ์…€์˜ ์ถœ๋ ฅ์—์„œ PcP_c ๊ฐ’๊ณผ IoUIoU ๊ฐ’์ด ์ œ์ผ ๋†’์€ ์…€์„ ์ œ์™ธํ•˜๊ณ  ๋‹ค๋ฅธ ์…€๋“ค์˜ ๊ฒฐ๊ณผ๋ฅผ ํƒˆ๋ฝ์‹œํ‚จ๋‹ค.

  • 19x19 ๊ฒฉ์ž ์ด๋ฏธ์ง€๋ฅผ ์˜ˆ๋กœ ๋“ค๊ณ , ํŽธ์˜์ƒ c1,c2,c3c_1, c_2, c_3 ๋ฅผ ์—†์• ๊ณ  ์ž๋™์ฐจ๋งŒ ์ธ์‹ํ•œ๋‹ค๊ณ  ๊ฐ€์ •ํ•œ๋‹ค.

[pcbxbybhbw]\begin{bmatrix}p_c\\b_x\\b_y\\b_h\\b_w\end{bmatrix}
  1. pcโ‰ค0.6p_c\le0.6 (ํ˜น์€ ํŠน์ • ์ž„๊ณ„ ๊ฐ’) ์ด๋ฉด ํ•ด๋‹น ์…€์„ ๋ฒ„๋ฆฐ๋‹ค.
  2. ํ•˜๋‚˜๋งŒ ๋‚จ์„ ๋•Œ ๊นŒ์ง€ ์•„๋ž˜ ๊ณผ์ •์„ ๋ฐ˜๋ณตํ•œ๋‹ค.
    1. pcp_c ๊ฐ€ ์ œ์ผ ํฐ ์…€์„ ์ทจํ•œ๋‹ค.
    2. IoUโ‰ฅ0.5IoU\ge0.5 ์ธ ์…€์„ ๋ฒ„๋ฆฐ๋‹ค.