| data: | |
| train_img_dir: 'dataset/train' | |
| train_mask_dir: '' | |
| val_img_dir: 'dataset/val' | |
| val_mask_dir: '' | |
| imgsz: 1024 | |
| augment: True | |
| num_workers: 8 | |
| cache: True | |
| aug_param: | |
| hsv: 0.3 | |
| mini_mosaic: 0.7 | |
| flip_lr: 0.5 | |
| neg: 0.3 | |
| size_range: [0.85, 1.1] | |
| rotate: 0.33 | |
| rotate_range: [-70, 70] | |
| save_dir: 'results' | |
| train: | |
| epochs: 160 | |
| linear_lr: False | |
| optimizer: 'adam' | |
| batch_size: 4 | |
| lr0: 0.01 | |
| lrf: 0.002 | |
| warm_up: True | |
| momentum: 0.937 | |
| weight_decay: 0.00002 | |
| warmup_epochs: 3.0 # warmup epochs (fractions ok) | |
| warmup_momentum: 0.8 # warmup initial momentum | |
| warmup_bias_lr: 0.1 # warmup initial bias lr | |
| eval_interval: 1 | |
| loss: 'bce' | |
| accumulation_steps: 4 | |
| model: | |
| weights: 'data/yolov5sblk.ckpt' | |
| unet_weights: 'data/unet_best.ckpt' | |
| db_weights: '' | |
| act: 'leaky' | |
| logger: | |
| type: 'wandb' | |
| run_id: '' | |
| project: 'TextDetectDB' | |
| resume: | |
| resume_training: False | |
| ckpt: '' | |