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

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

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



Watch: ๊ต์œก ๋ฐฉ๋ฒ• ์ด๋ฏธ์ง€ ๋ถ„๋ฅ˜ Caltech-256 ๋ฐ์ดํ„ฐ์„ธํŠธ์™€ Ultralytics HUB๋ฅผ ์‚ฌ์šฉํ•œ ๋ชจ๋ธ๋ง

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

  • 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("yolo11n-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 classify train data=caltech256 model=yolo11n-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 ๋ฐ์ดํ„ฐ ์„ธํŠธ ์›น ์‚ฌ์ดํŠธ๋ฅผ ๋ฐฉ๋ฌธํ•˜์„ธ์š”.

์ž์ฃผ ๋ฌป๋Š” ์งˆ๋ฌธ

Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” ๋ฌด์—‡์ด๋ฉฐ ๋จธ์‹  ๋Ÿฌ๋‹์— ์ค‘์š”ํ•œ ์ด์œ ๋Š” ๋ฌด์—‡์ธ๊ฐ€์š”?

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

Python ๋˜๋Š” CLI ์„ ์‚ฌ์šฉํ•˜์—ฌ Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ์—์„œ YOLO ๋ชจ๋ธ์„ ํ›ˆ๋ จํ•˜๋ ค๋ฉด ์–ด๋–ป๊ฒŒ ํ•ด์•ผ ํ•˜๋‚˜์š”?

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

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

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n-cls.pt")  # load a pretrained model

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

Caltech-256 ๋ฐ์ดํ„ฐ ์ง‘ํ•ฉ์˜ ๊ฐ€์žฅ ์ผ๋ฐ˜์ ์ธ ์‚ฌ์šฉ ์‚ฌ๋ก€๋Š” ๋ฌด์—‡์ธ๊ฐ€์š”?

Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” ๋‹ค์Œ๊ณผ ๊ฐ™์€ ๋‹ค์–‘ํ•œ ๋ฌผ์ฒด ์ธ์‹ ์ž‘์—…์— ๋„๋ฆฌ ์‚ฌ์šฉ๋ฉ๋‹ˆ๋‹ค:

๋‹ค์–‘์„ฑ๊ณผ ํฌ๊ด„์ ์ธ ์ฃผ์„์œผ๋กœ ๋จธ์‹  ๋Ÿฌ๋‹ ๋ฐ ์ปดํ“จํ„ฐ ๋น„์ „ ๋ถ„์•ผ์˜ ์—ฐ๊ตฌ ๊ฐœ๋ฐœ์— ์ด์ƒ์ ์ž…๋‹ˆ๋‹ค.

๊ต์œก ๋ฐ ํ…Œ์ŠคํŠธ๋ฅผ ์œ„ํ•ด Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” ์–ด๋–ป๊ฒŒ ๊ตฌ์กฐํ™”๋˜๊ณ  ๋ถ„ํ• ๋˜๋‚˜์š”?

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

Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ์˜ ๋ชจ๋ธ ํ•™์Šต์— Ultralytics YOLO ์„ ์‚ฌ์šฉํ•ด์•ผ ํ•˜๋Š” ์ด์œ ๋Š” ๋ฌด์—‡์ธ๊ฐ€์š”?

Ultralytics YOLO ๋ชจ๋ธ์€ Caltech-256 ๋ฐ์ดํ„ฐ ์„ธํŠธ์— ๋Œ€ํ•œ ๊ต์œก์— ๋ช‡ ๊ฐ€์ง€ ์ด์ ์„ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค:

  • ๋†’์€ ์ •ํ™•๋„: YOLO ๋ชจ๋ธ์€ ๋ฌผ์ฒด ๊ฐ์ง€ ์ž‘์—…์—์„œ ์ตœ์ฒจ๋‹จ ์„ฑ๋Šฅ์œผ๋กœ ์ž˜ ์•Œ๋ ค์ ธ ์žˆ์Šต๋‹ˆ๋‹ค.
  • ์†๋„: ์‹ค์‹œ๊ฐ„ ์ถ”๋ก  ๊ธฐ๋Šฅ์„ ์ œ๊ณตํ•˜๋ฏ€๋กœ ๋น ๋ฅธ ์˜ˆ์ธก์ด ํ•„์š”ํ•œ ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜์— ์ ํ•ฉํ•ฉ๋‹ˆ๋‹ค.
  • ์‚ฌ์šฉ ํŽธ์˜์„ฑ: Ultralytics HUB๋ฅผ ์‚ฌ์šฉํ•˜๋ฉด ๋Œ€๊ทœ๋ชจ ์ฝ”๋”ฉ ์—†์ด๋„ ๋ชจ๋ธ์„ ํ•™์Šต, ๊ฒ€์ฆ ๋ฐ ๋ฐฐํฌํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
  • ์‚ฌ์ „ ํ•™์Šต๋œ ๋ชจ๋ธ: ๋‹ค์Œ๊ณผ ๊ฐ™์ด ์‚ฌ์ „ ํ•™์Šต๋œ ๋ชจ๋ธ๋ถ€ํ„ฐ ์‹œ์ž‘ yolo11n-cls.pt๋ฅผ ์‚ฌ์šฉํ•˜๋ฉด ํ›ˆ๋ จ ์‹œ๊ฐ„์„ ํฌ๊ฒŒ ์ค„์ด๊ณ  ๋ชจ๋ธ์„ ๊ฐœ์„ ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์ •ํ™•์„ฑ.

์ž์„ธํ•œ ๋‚ด์šฉ์€ ์ข…ํ•ฉ ๊ต์œก ๊ฐ€์ด๋“œ๋ฅผ ์ฐธ์กฐํ•˜์„ธ์š”.

๐Ÿ“…1 ๋…„ ์ „ ์ƒ์„ฑ๋จ โœ๏ธ 1๊ฐœ์›” ์ „ ์—…๋ฐ์ดํŠธ๋จ

๋Œ“๊ธ€