์ฝ˜ํ…์ธ ๋กœ ๊ฑด๋„ˆ๋›ฐ๊ธฐ

Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ

Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” ๋ฌผ์ฒด ๋ถ„๋ฅ˜ ์ž‘์—…์— ์‚ฌ์šฉ๋˜๋Š” ๊ด‘๋ฒ”์œ„ํ•œ ์ด๋ฏธ์ง€ ๋ชจ์Œ์ž…๋‹ˆ๋‹ค. 257๊ฐœ์˜ ์นดํ…Œ๊ณ ๋ฆฌ(256๊ฐœ์˜ ๊ฐ์ฒด ์นดํ…Œ๊ณ ๋ฆฌ์™€ 1๊ฐœ์˜ ๋ฐฐ๊ฒฝ ์นดํ…Œ๊ณ ๋ฆฌ)๋กœ ๋‚˜๋‰œ ์•ฝ 30,000๊ฐœ์˜ ์ด๋ฏธ์ง€๊ฐ€ ํฌํ•จ๋˜์–ด ์žˆ์Šต๋‹ˆ๋‹ค. ์ด ์ด๋ฏธ์ง€๋“ค์€ ๋ฌผ์ฒด ์ธ์‹ ์•Œ๊ณ ๋ฆฌ์ฆ˜์„ ์œ„ํ•œ ๋„์ „์ ์ด๊ณ  ๋‹ค์–‘ํ•œ ๋ฒค์น˜๋งˆํฌ๋ฅผ ์ œ๊ณตํ•˜๊ธฐ ์œ„ํ•ด ์‹ ์ค‘ํ•˜๊ฒŒ ํ๋ ˆ์ดํŒ…๋˜๊ณ  ์ฃผ์„์ด ๋‹ฌ๋ ธ์Šต๋‹ˆ๋‹ค.

์ฃผ์š” ๊ธฐ๋Šฅ

  • Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” 257๊ฐœ์˜ ์นดํ…Œ๊ณ ๋ฆฌ๋กœ ๋‚˜๋‰œ ์•ฝ 30,000๊ฐœ์˜ ์ปฌ๋Ÿฌ ์ด๋ฏธ์ง€๋กœ ๊ตฌ์„ฑ๋˜์–ด ์žˆ์Šต๋‹ˆ๋‹ค.
  • ๊ฐ ์นดํ…Œ๊ณ ๋ฆฌ์—๋Š” ์ตœ์†Œ 80๊ฐœ์˜ ์ด๋ฏธ์ง€๊ฐ€ ํฌํ•จ๋ฉ๋‹ˆ๋‹ค.
  • ์นดํ…Œ๊ณ ๋ฆฌ์—๋Š” ๋™๋ฌผ, ์ฐจ๋Ÿ‰, ๊ฐ€์ •์šฉํ’ˆ, ์‚ฌ๋žŒ ๋“ฑ ๋‹ค์–‘ํ•œ ์‹ค์ œ ์‚ฌ๋ฌผ์ด ํฌํ•จ๋ฉ๋‹ˆ๋‹ค.
  • ์ด๋ฏธ์ง€์˜ ํฌ๊ธฐ์™€ ํ•ด์ƒ๋„๋Š” ๋‹ค์–‘ํ•ฉ๋‹ˆ๋‹ค.
  • Caltech-256์€ ๋จธ์‹  ๋Ÿฌ๋‹ ๋ถ„์•ผ, ํŠนํžˆ ๊ฐ์ฒด ์ธ์‹ ์ž‘์—…์˜ ๊ต์œก ๋ฐ ํ…Œ์ŠคํŠธ์— ๋„๋ฆฌ ์‚ฌ์šฉ๋ฉ๋‹ˆ๋‹ค.

๋ฐ์ดํ„ฐ ์„ธํŠธ ๊ตฌ์กฐ

Caltech-101๊ณผ ๋งˆ์ฐฌ๊ฐ€์ง€๋กœ, Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” ํ›ˆ๋ จ ์„ธํŠธ์™€ ํ…Œ์ŠคํŠธ ์„ธํŠธ ์‚ฌ์ด์— ๊ณต์‹์ ์ธ ๋ถ„ํ• ์ด ์—†์Šต๋‹ˆ๋‹ค. ์‚ฌ์šฉ์ž๋Š” ์ผ๋ฐ˜์ ์œผ๋กœ ํŠน์ • ํ•„์š”์— ๋”ฐ๋ผ ์ž์ฒด์ ์œผ๋กœ ๋ถ„ํ• ์„ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค. ์ผ๋ฐ˜์ ์œผ๋กœ ์ด๋ฏธ์ง€์˜ ๋ฌด์ž‘์œ„ ํ•˜์œ„ ์ง‘ํ•ฉ์„ ํŠธ๋ ˆ์ด๋‹์— ์‚ฌ์šฉํ•˜๊ณ  ๋‚˜๋จธ์ง€ ์ด๋ฏธ์ง€๋ฅผ ํ…Œ์ŠคํŠธ์— ์‚ฌ์šฉํ•˜๋Š” ๊ฒƒ์ด ์ผ๋ฐ˜์ ์ž…๋‹ˆ๋‹ค.

์• ํ”Œ๋ฆฌ์ผ€์ด์…˜

Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” ์ปจ๋ณผ๋ฃจ์…˜ ์‹ ๊ฒฝ๋ง(CNN), ์„œํฌํŠธ ๋ฒกํ„ฐ ๋จธ์‹ (SVM) ๋ฐ ๊ธฐํƒ€ ๋‹ค์–‘ํ•œ ๋จธ์‹  ๋Ÿฌ๋‹ ์•Œ๊ณ ๋ฆฌ์ฆ˜๊ณผ ๊ฐ™์€ ๊ฐ์ฒด ์ธ์‹ ์ž‘์—…์—์„œ ๋”ฅ ๋Ÿฌ๋‹ ๋ชจ๋ธ์„ ํ›ˆ๋ จํ•˜๊ณ  ํ‰๊ฐ€ํ•˜๋Š” ๋ฐ ๊ด‘๋ฒ”์œ„ํ•˜๊ฒŒ ์‚ฌ์šฉ๋ฉ๋‹ˆ๋‹ค. ๋‹ค์–‘ํ•œ ์นดํ…Œ๊ณ ๋ฆฌ์™€ ๊ณ ํ’ˆ์งˆ ์ด๋ฏธ์ง€๋กœ ๊ตฌ์„ฑ๋˜์–ด ์žˆ์–ด ๋จธ์‹  ๋Ÿฌ๋‹ ๋ฐ ์ปดํ“จํ„ฐ ๋น„์ „ ๋ถ„์•ผ์˜ ์—ฐ๊ตฌ ๊ฐœ๋ฐœ์— ๋งค์šฐ ๊ท€์ค‘ํ•œ ๋ฐ์ดํ„ฐ ์„ธํŠธ์ž…๋‹ˆ๋‹ค.

์‚ฌ์šฉ๋ฒ•

๋‹ค์Œ ์ฝ”๋“œ ์กฐ๊ฐ์„ ์‚ฌ์šฉํ•˜์—ฌ 100๊ฐœ ์‹œ๋Œ€์— ๋Œ€ํ•œ Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ์— ๋Œ€ํ•ด YOLO ๋ชจ๋ธ์„ ํ›ˆ๋ จํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์‚ฌ์šฉ ๊ฐ€๋Šฅํ•œ ์ธ์ˆ˜์˜ ์ „์ฒด ๋ชฉ๋ก์€ ๋ชจ๋ธ ํ›ˆ๋ จ ํŽ˜์ด์ง€๋ฅผ ์ฐธ์กฐํ•˜์„ธ์š”.

์—ด์ฐจ ์˜ˆ์‹œ

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n-cls.pt')  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data='caltech256', epochs=100, imgsz=416)
# Start training from a pretrained *.pt model
yolo detect train data=caltech256 model=yolov8n-cls.pt epochs=100 imgsz=416

์ƒ˜ํ”Œ ์ด๋ฏธ์ง€ ๋ฐ ์ฃผ์„

Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ์—๋Š” ๋‹ค์–‘ํ•œ ๋ฌผ์ฒด์˜ ๊ณ ํ’ˆ์งˆ ์ปฌ๋Ÿฌ ์ด๋ฏธ์ง€๊ฐ€ ํฌํ•จ๋˜์–ด ์žˆ์–ด ๋ฌผ์ฒด ์ธ์‹ ์ž‘์—…์„ ์œ„ํ•œ ํฌ๊ด„์ ์ธ ๋ฐ์ดํ„ฐ ์„ธํŠธ๋ฅผ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค. ๋‹ค์Œ์€ ๋ฐ์ดํ„ฐ ์„ธํŠธ์˜ ์ด๋ฏธ์ง€ ๋ช‡ ๊ฐ€์ง€ ์˜ˆ์‹œ์ž…๋‹ˆ๋‹ค(ํฌ๋ ˆ๋”ง ์ œ๊ณต):

๋ฐ์ดํ„ฐ ์„ธํŠธ ์ƒ˜ํ”Œ ์ด๋ฏธ์ง€

์ด ์˜ˆ๋Š” Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ์— ํฌํ•จ๋œ ๊ฐ์ฒด์˜ ๋‹ค์–‘์„ฑ๊ณผ ๋ณต์žก์„ฑ์„ ๋ณด์—ฌ์ฃผ๋ฉฐ, ๊ฐ•๋ ฅํ•œ ๊ฐ์ฒด ์ธ์‹ ๋ชจ๋ธ์„ ํ›ˆ๋ จํ•˜๊ธฐ ์œ„ํ•ด ๋‹ค์–‘ํ•œ ๋ฐ์ดํ„ฐ ์„ธํŠธ๊ฐ€ ์ค‘์š”ํ•˜๋‹ค๋Š” ์ ์„ ๊ฐ•์กฐํ•ฉ๋‹ˆ๋‹ค.

์ธ์šฉ ๋ฐ ๊ฐ์‚ฌ

์—ฐ๊ตฌ ๋˜๋Š” ๊ฐœ๋ฐœ ์ž‘์—…์—์„œ Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ๋ฅผ ์‚ฌ์šฉํ•˜๋Š” ๊ฒฝ์šฐ ๋‹ค์Œ ๋…ผ๋ฌธ์„ ์ธ์šฉํ•ด ์ฃผ์„ธ์š”:

@article{griffin2007caltech,
         title={Caltech-256 object category dataset},
         author={Griffin, Gregory and Holub, Alex and Perona, Pietro},
         year={2007}
}

๋จธ์‹ ๋Ÿฌ๋‹ ๋ฐ ์ปดํ“จํ„ฐ ๋น„์ „ ์—ฐ๊ตฌ ์ปค๋ฎค๋‹ˆํ‹ฐ๋ฅผ ์œ„ํ•œ ๊ท€์ค‘ํ•œ ๋ฆฌ์†Œ์Šค์ธ Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ๋ฅผ ๋งŒ๋“ค๊ณ  ์œ ์ง€ ๊ด€๋ฆฌํ•ด ์ฃผ์‹  Gregory Griffin, Alex Holub, Pietro Perona์—๊ฒŒ ๊ฐ์‚ฌ์˜ ๋ง์”€์„ ์ „ํ•ฉ๋‹ˆ๋‹ค. ์ž์„ธํ•œ ๋‚ด์šฉ์€

Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ์™€ ์ œ์ž‘์ž์— ๋Œ€ํ•œ ์ž์„ธํ•œ ๋‚ด์šฉ์€ Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ ์›น์‚ฌ์ดํŠธ๋ฅผ ๋ฐฉ๋ฌธํ•˜์„ธ์š”.



2023-11-12 ์ƒ์„ฑ, 2023-11-22 ์—…๋ฐ์ดํŠธ๋จ
์ž‘์„ฑ์ž: glenn-jocher (3)

๋Œ“๊ธ€