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

์„œ๋ช… ํƒ์ง€ ๋ฐ์ดํ„ฐ ์„ธํŠธ

์ด ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” ๋ฌธ์„œ ๋‚ด์—์„œ ์‚ฌ๋žŒ์ด ์ง์ ‘ ์ž‘์„ฑํ•œ ์„œ๋ช…์„ ๊ฐ์ง€ํ•˜๋Š” ๋ฐ ์ค‘์ ์„ ๋‘ก๋‹ˆ๋‹ค. ์—ฌ๊ธฐ์—๋Š” ์ฃผ์„์ด ๋‹ฌ๋ฆฐ ์„œ๋ช…์ด ์žˆ๋Š” ๋‹ค์–‘ํ•œ ๋ฌธ์„œ ์œ ํ˜•์ด ํฌํ•จ๋˜์–ด ์žˆ์–ด ๋ฌธ์„œ ๊ฒ€์ฆ ๋ฐ ์‚ฌ๊ธฐ ํƒ์ง€ ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜์— ์œ ์šฉํ•œ ์ธ์‚ฌ์ดํŠธ๋ฅผ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค. ์ปดํ“จํ„ฐ ๋น„์ „ ์•Œ๊ณ ๋ฆฌ์ฆ˜ ํ›ˆ๋ จ์— ํ•„์ˆ˜์ ์ธ ์ด ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” ๋‹ค์–‘ํ•œ ๋ฌธ์„œ ํ˜•์‹์˜ ์„œ๋ช…์„ ์‹๋ณ„ํ•˜๋Š” ๋ฐ ๋„์›€์ด ๋˜๋ฉฐ, ๋ฌธ์„œ ๋ถ„์„์— ๋Œ€ํ•œ ์—ฐ๊ตฌ ๋ฐ ์‹ค์ œ ์ ์šฉ์„ ์ง€์›ํ•ฉ๋‹ˆ๋‹ค.

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

์„œ๋ช… ๊ฐ์ง€ ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” ์„ธ ๊ฐœ์˜ ํ•˜์œ„ ์ง‘ํ•ฉ์œผ๋กœ ๋‚˜๋‰ฉ๋‹ˆ๋‹ค:

  • ํŠธ๋ ˆ์ด๋‹ ์„ธํŠธ: 143๊ฐœ์˜ ์ด๋ฏธ์ง€์™€ ๊ฐ๊ฐ ํ•ด๋‹น ์ฃผ์„์ด ํฌํ•จ๋˜์–ด ์žˆ์Šต๋‹ˆ๋‹ค.
  • ์œ ํšจ์„ฑ ๊ฒ€์‚ฌ ์„ธํŠธ: 35๊ฐœ์˜ ์ด๋ฏธ์ง€์™€ ๊ฐ๊ฐ ์Œ์„ ์ด๋ฃจ๋Š” ์ฃผ์„์„ ํฌํ•จํ•ฉ๋‹ˆ๋‹ค.

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

์ด ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” ๊ฐ์ฒด ๊ฐ์ง€, ๊ฐ์ฒด ์ถ”์ , ๋ฌธ์„œ ๋ถ„์„๊ณผ ๊ฐ™์€ ๋‹ค์–‘ํ•œ ์ปดํ“จํ„ฐ ๋น„์ „ ์ž‘์—…์— ์ ์šฉํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ํŠนํžˆ ๋ฌธ์„œ์—์„œ ์„œ๋ช…์„ ์‹๋ณ„ํ•˜๋Š” ๋ชจ๋ธ์„ ํ›ˆ๋ จํ•˜๊ณ  ํ‰๊ฐ€ํ•˜๋Š” ๋ฐ ์‚ฌ์šฉํ•  ์ˆ˜ ์žˆ์œผ๋ฉฐ, ์ด๋Š” ๋ฌธ์„œ ๊ฒ€์ฆ, ์‚ฌ๊ธฐ ํƒ์ง€, ๊ธฐ๋ก ์—ฐ๊ตฌ ๋“ฑ์— ์‘์šฉ๋  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ๋˜ํ•œ ํ•™์ƒ๊ณผ ์—ฐ๊ตฌ์ž๊ฐ€ ๋‹ค์–‘ํ•œ ๋ฌธ์„œ ์œ ํ˜•์—์„œ ์„œ๋ช…์˜ ํŠน์„ฑ๊ณผ ๋™์ž‘์„ ์—ฐ๊ตฌํ•˜๊ณ  ์ดํ•ดํ•  ์ˆ˜ ์žˆ๋„๋ก ๊ต์œก ๋ชฉ์ ์œผ๋กœ๋„ ์œ ์šฉํ•œ ๋ฆฌ์†Œ์Šค๋กœ ํ™œ์šฉํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

๋ฐ์ดํ„ฐ ์„ธํŠธ YAML

YAML(๋˜ ๋‹ค๋ฅธ ๋งˆํฌ์—… ์–ธ์–ด) ํŒŒ์ผ์€ ๊ฒฝ๋กœ ๋ฐ ํด๋ž˜์Šค ์ •๋ณด๋ฅผ ํฌํ•จํ•œ ๋ฐ์ดํ„ฐ ์„ธํŠธ ๊ตฌ์„ฑ์„ ์ •์˜ํ•ฉ๋‹ˆ๋‹ค. ์„œ๋ช… ๊ฐ์ง€ ๋ฐ์ดํ„ฐ ์„ธํŠธ์˜ ๊ฒฝ์šฐ, ์„œ๋ช… ๊ฐ์ง€ ๋ฐ์ดํ„ฐ ์„ธํŠธ์˜ signature.yaml ํŒŒ์ผ์€ ๋‹ค์Œ ์œ„์น˜์— ์žˆ์Šต๋‹ˆ๋‹ค. https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/signature.yaml.

ultralytics/cfg/๋ฐ์ดํ„ฐ์„ธํŠธ/์„œ๋ช….yaml

# Ultralytics YOLO ๐Ÿš€, AGPL-3.0 license
# Signature dataset by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/detect/signature/
# Example usage: yolo train data=signature.yaml
# parent
# โ”œโ”€โ”€ ultralytics
# โ””โ”€โ”€ datasets
#     โ””โ”€โ”€ signature  โ† downloads here (11.2 MB)

# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: ../datasets/signature # dataset root dir
train: train/images # train images (relative to 'path') 143 images
val: valid/images # val images (relative to 'path') 35 images

# Classes
names:
  0: signature

# Download script/URL (optional)
download: https://ultralytics.com/assets/signature.zip

์‚ฌ์šฉ๋ฒ•

์ด๋ฏธ์ง€ ํฌ๊ธฐ๊ฐ€ 640์ธ 100๊ฐœ์˜ ์—ํฌํฌ์— ๋Œ€ํ•œ ์„œ๋ช… ๊ฐ์ง€ ๋ฐ์ดํ„ฐ ์„ธํŠธ์— ๋Œ€ํ•ด YOLOv8n ๋ชจ๋ธ์„ ํ›ˆ๋ จํ•˜๋ ค๋ฉด ์ œ๊ณต๋œ ์ฝ”๋“œ ์ƒ˜ํ”Œ์„ ์‚ฌ์šฉํ•˜์„ธ์š”. ์‚ฌ์šฉ ๊ฐ€๋Šฅํ•œ ํŒŒ๋ผ๋ฏธํ„ฐ์˜ ์ „์ฒด ๋ชฉ๋ก์€ ๋ชจ๋ธ์˜ ํ•™์Šต ํŽ˜์ด์ง€๋ฅผ ์ฐธ์กฐํ•˜์„ธ์š”.

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

from ultralytics import YOLO

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

# Train the model
results = model.train(data="signature.yaml", epochs=100, imgsz=640)
# Start training from a pretrained *.pt model
yolo detect train data=signature.yaml model=yolov8n.pt epochs=100 imgsz=640

์ถ”๋ก  ์˜ˆ์ œ

from ultralytics import YOLO

# Load a model
model = YOLO("path/to/best.pt")  # load a signature-detection fine-tuned model

# Inference using the model
results = model.predict("https://ultralytics.com/assets/signature-s.mp4", conf=0.75)
# Start prediction with a finetuned *.pt model
yolo detect predict model='path/to/best.pt' imgsz=640 source="https://ultralytics.com/assets/signature-s.mp4" conf=0.75

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

์„œ๋ช… ๊ฐ์ง€ ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” ๋‹ค์–‘ํ•œ ๋ฌธ์„œ ์œ ํ˜•๊ณผ ์ฃผ์„์ด ๋‹ฌ๋ฆฐ ์„œ๋ช…์„ ๋ณด์—ฌ์ฃผ๋Š” ๋‹ค์–‘ํ•œ ์ด๋ฏธ์ง€๋กœ ๊ตฌ์„ฑ๋˜์–ด ์žˆ์Šต๋‹ˆ๋‹ค. ์•„๋ž˜๋Š” ๋ฐ์ดํ„ฐ ์„ธํŠธ์˜ ์ด๋ฏธ์ง€ ์˜ˆ์‹œ์ด๋ฉฐ, ๊ฐ ์ด๋ฏธ์ง€์—๋Š” ํ•ด๋‹น ์ฃผ์„์ด ์ฒจ๋ถ€๋˜์–ด ์žˆ์Šต๋‹ˆ๋‹ค.

์„œ๋ช… ๊ฐ์ง€ ๋ฐ์ดํ„ฐ ์„ธํŠธ ์ƒ˜ํ”Œ ์ด๋ฏธ์ง€

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

์ด ์˜ˆ๋Š” ์„œ๋ช… ๊ฐ์ง€ ๋ฐ์ดํ„ฐ ์„ธํŠธ์— ํฌํ•จ๋œ ์ด๋ฏธ์ง€์˜ ๋‹ค์–‘์„ฑ๊ณผ ๋ณต์žก์„ฑ์„ ๋ณด์—ฌ์ฃผ๋ฉฐ, ํ•™์Šต ๊ณผ์ •์—์„œ ๋ชจ์ž์ดํ‚น์„ ํฌํ•จํ•  ๋•Œ ์–ป์„ ์ˆ˜ ์žˆ๋Š” ์ด์ ์„ ๊ฐ•์กฐํ•ฉ๋‹ˆ๋‹ค.

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

์ด ๋ฐ์ดํ„ฐ ์„ธํŠธ๋Š” AGPL-3.0 ๋ผ์ด์„ ์Šค์— ๋”ฐ๋ผ ๊ณต๊ฐœ๋˜์—ˆ์Šต๋‹ˆ๋‹ค.



Created 2024-05-22, Updated 2024-06-02
Authors: glenn-jocher (1), RizwanMunawar (1)

๋Œ“๊ธ€