рдХреЗ рд▓рд┐рдП рд╕рдВрджрд░реНрдн ultralytics/models/yolo/world/train_world.py
рдиреЛрдЯ
рдпрд╣ рдлрд╝рд╛рдЗрд▓ рдпрд╣рд╛рдБ рдЙрдкрд▓рдмреНрдз рд╣реИ https://github.com/ultralytics/ultralytics/рдмреВрдБрдж/рдореБрдЦреНрдп/ultralytics/рдореЙрдбрд▓/yolo/рджреБрдирд┐рдпрд╛/train_world.py рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░реЗрдВред рдпрджрд┐ рдЖрдк рдХреЛрдИ рд╕рдорд╕реНрдпрд╛ рджреЗрдЦрддреЗ рд╣реИрдВ рддреЛ рдХреГрдкрдпрд╛ рдкреБрд▓ рдЕрдиреБрд░реЛрдз рдХрд╛ рдпреЛрдЧрджрд╛рди рдХрд░рдХреЗ рдЗрд╕реЗ рдареАрдХ рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░реЗрдВ ЁЯЫая╕Пред ЁЯЩП рдзрдиреНрдпрд╡рд╛рдж !
ultralytics.models.yolo.world.train_world.WorldTrainerFromScratch
рдХрд╛ рд░реВрдк: 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 рдСрдмреНрдЬреЗрдХреНрдЯ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/yolo/world/train_world.py
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-рд╡рд░реНрд▓реНрдб рдореЙрдбрд▓ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/yolo/world/train_world.py
get_dataset()
рдпрджрд┐ рдпрд╣ рдореМрдЬреВрдж рд╣реИ рддреЛ рдбреЗрдЯрд╛ рдбрд┐рдХреНрдЯ рд╕реЗ рдЯреНрд░реЗрди, рд╡реИрд▓ рдкрде рдкреНрд░рд╛рдкреНрдд рдХрд░реЗрдВред
рдпрджрд┐ рдбреЗрдЯрд╛ рд╕реНрд╡рд░реВрдк рдкрд╣рдЪрд╛рдирд╛ рдирд╣реАрдВ рдЧрдпрд╛ рд╣реИ рддреЛ рдХреЛрдИ рдирд╣реАрдВ рд▓реМрдЯрд╛рддрд╛ рд╣реИ.
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/yolo/world/train_world.py
2024-03-31 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-05-08
рд▓реЗрдЦрдХ: рдмреБрд░рд╣рд╛рди-рдХреНрдпреВ (1), рд▓рд╛рдлрд┐рдВрдЧ-рдХреНрдпреВ (1)