106 lines
1.5 KiB
YAML
106 lines
1.5 KiB
YAML
task: detect
|
|
mode: train
|
|
model: yolo12s.yaml
|
|
data: datasets/data.yaml
|
|
epochs: 600
|
|
time: null
|
|
patience: 100
|
|
batch: 32
|
|
imgsz: 640
|
|
save: true
|
|
save_period: 50
|
|
cache: disk
|
|
device: '0'
|
|
workers: 4
|
|
project: run
|
|
name: train
|
|
exist_ok: false
|
|
pretrained: true
|
|
optimizer: SGD
|
|
verbose: true
|
|
seed: 0
|
|
deterministic: true
|
|
single_cls: false
|
|
rect: false
|
|
cos_lr: false
|
|
close_mosaic: 10
|
|
resume: false
|
|
amp: true
|
|
fraction: 1.0
|
|
profile: false
|
|
freeze: null
|
|
multi_scale: false
|
|
overlap_mask: true
|
|
mask_ratio: 4
|
|
dropout: 0.0
|
|
val: true
|
|
split: val
|
|
save_json: false
|
|
conf: null
|
|
iou: 0.7
|
|
max_det: 300
|
|
half: false
|
|
dnn: false
|
|
plots: true
|
|
source: null
|
|
vid_stride: 1
|
|
stream_buffer: false
|
|
visualize: false
|
|
augment: false
|
|
agnostic_nms: false
|
|
classes: null
|
|
retina_masks: false
|
|
embed: null
|
|
show: false
|
|
save_frames: false
|
|
save_txt: false
|
|
save_conf: false
|
|
save_crop: false
|
|
show_labels: true
|
|
show_conf: true
|
|
show_boxes: true
|
|
line_width: null
|
|
format: torchscript
|
|
keras: false
|
|
optimize: false
|
|
int8: false
|
|
dynamic: false
|
|
simplify: true
|
|
opset: null
|
|
workspace: null
|
|
nms: false
|
|
lr0: 0.001
|
|
lrf: 0.01
|
|
momentum: 0.937
|
|
weight_decay: 0.0005
|
|
warmup_epochs: 3.0
|
|
warmup_momentum: 0.8
|
|
warmup_bias_lr: 0.1
|
|
box: 7.5
|
|
cls: 0.5
|
|
dfl: 1.5
|
|
pose: 12.0
|
|
kobj: 1.0
|
|
nbs: 64
|
|
hsv_h: 0.015
|
|
hsv_s: 0.7
|
|
hsv_v: 0.4
|
|
degrees: 30
|
|
translate: 0.1
|
|
scale: 0.5
|
|
shear: 15
|
|
perspective: 0.0001
|
|
flipud: 0.5
|
|
fliplr: 0.5
|
|
bgr: 0.0
|
|
mosaic: 1.0
|
|
mixup: 0.2
|
|
cutmix: 0.0
|
|
copy_paste: 0.3
|
|
copy_paste_mode: flip
|
|
auto_augment: randaugment
|
|
erasing: 0.4
|
|
cfg: null
|
|
tracker: botsort.yaml
|
|
save_dir: run\train
|