์ฐธ์กฐ ultralytics/models/yolo/world/train_world.py
์ฐธ๊ณ
์ด ํ์ผ์ https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ yolo/world/train_world .py์์ ํ์ธํ ์ ์์ต๋๋ค. ๋ฌธ์ ๋ฅผ ๋ฐ๊ฒฌํ๋ฉด ํ ๋ฆฌํ์คํธ (๐ ๏ธ) ๋ฅผ ํตํด ๋ฌธ์ ๋ฅผ ํด๊ฒฐํ๋๋ก ๋์์ฃผ์ธ์. ๊ฐ์ฌํฉ๋๋ค ๐!
ultralytics.models.yolo.world.train_world.WorldTrainerFromScratch
๋ฒ ์ด์ค: WorldTrainer
์คํ ์ธํธ ๋ฐ์ดํฐ ์ธํธ์์ ์๋ ๋ชจ๋ธ์ ์ฒ์๋ถํฐ ํ๋ จํ๊ธฐ ์ํ WorldTrainer ํด๋์ค๋ฅผ ํ์ฅํ ํด๋์ค์ ๋๋ค.
์
from ultralytics.models.yolo.world.train_world import WorldTrainerFromScratch
from ultralytics import YOLOWorld
data = dict(
train=dict(
yolo_data=["Objects365.yaml"],
grounding_data=[
dict(
img_path="../datasets/flickr30k/images",
json_file="../datasets/flickr30k/final_flickr_separateGT_train.json",
),
dict(
img_path="../datasets/GQA/images",
json_file="../datasets/GQA/final_mixed_train_no_coco.json",
),
],
),
val=dict(yolo_data=["lvis.yaml"]),
)
model = YOLOWorld("yolov8s-worldv2.yaml")
model.train(data=data, trainer=WorldTrainerFromScratch)
์ ์์ค ์ฝ๋ ultralytics/models/yolo/world/train_world.py
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
|
__init__(cfg=DEFAULT_CFG, overrides=None, _callbacks=None)
์ฃผ์ด์ง ์ธ์๋ฅผ ์ฌ์ฉํ์ฌ WorldTrainer ๊ฐ์ฒด๋ฅผ ์ด๊ธฐํํฉ๋๋ค.
build_dataset(img_path, mode='train', batch=None)
YOLO ๋ฐ์ดํฐ ์งํฉ์ ๊ตฌ์ถํฉ๋๋ค.
๋งค๊ฐ๋ณ์:
์ด๋ฆ | ์ ํ | ์ค๋ช | ๊ธฐ๋ณธ๊ฐ |
---|---|---|---|
img_path |
List[str] | str
|
์ด๋ฏธ์ง๊ฐ ํฌํจ๋ ํด๋์ ๊ฒฝ๋ก์ ๋๋ค. |
ํ์ |
mode |
str
|
|
'train'
|
batch |
int
|
๋ฐฐ์น์ ํฌ๊ธฐ, ์ด๊ฒ์ ๋ค์์ ์ํ ๊ฒ์
๋๋ค. |
None
|
์ ์์ค ์ฝ๋ ultralytics/models/yolo/world/train_world.py
final_eval()
๊ฐ์ฒด ๊ฐ์ง๋ฅผ ์ํ ์ต์ข ํ๊ฐ ๋ฐ ์ ํจ์ฑ ๊ฒ์ฌ๋ฅผ ์ํํฉ๋๋ค YOLO-World ๋ชจ๋ธ.
์ ์์ค ์ฝ๋ ultralytics/models/yolo/world/train_world.py
get_dataset()
๋ฐ์ดํฐ ๋์ ๋๋ฆฌ์์ ๊ธฐ์ฐจ, ๊ฐ ๊ฒฝ๋ก๊ฐ ์กด์ฌํ๋ ๊ฒฝ์ฐ ๊ฐ์ ธ์ต๋๋ค.
๋ฐ์ดํฐ ํ์์ด ์ธ์๋์ง ์์ผ๋ฉด None์ ๋ฐํํฉ๋๋ค.