| { | |
| "model": { | |
| "type": "FAST", | |
| "backbone": { | |
| "type": "fast_backbone", | |
| "config": "config/fast/nas-configs/fast_small.config" | |
| }, | |
| "neck": { | |
| "type": "fast_neck", | |
| "config": "config/fast/nas-configs/fast_small.config" | |
| }, | |
| "detection_head": { | |
| "type": "fast_head", | |
| "config": "config/fast/nas-configs/fast_small.config", | |
| "pooling_size": 9, | |
| "loss_text": { | |
| "type": "DiceLoss", | |
| "loss_weight": 0.5 | |
| }, | |
| "loss_kernel": { | |
| "type": "DiceLoss", | |
| "loss_weight": 1.0 | |
| }, | |
| "loss_emb": { | |
| "type": "EmbLoss_v1", | |
| "feature_dim": 4, | |
| "loss_weight": 0.25 | |
| } | |
| } | |
| }, | |
| "repeat_times": 10, | |
| "data": { | |
| "batch_size": 16, | |
| "train": { | |
| "type": "FAST_IC17MLT", | |
| "split": "train", | |
| "is_transform": true, | |
| "img_size": 640, | |
| "short_size": 640, | |
| "pooling_size": 9, | |
| "read_type": "cv2", | |
| "repeat_times": 10 | |
| }, | |
| "test": { | |
| "type": "FAST_IC17MLT", | |
| "split": "test", | |
| "short_size": 640, | |
| "read_type": "cv2" | |
| } | |
| }, | |
| "train_cfg": { | |
| "lr": 0.001, | |
| "schedule": "polylr", | |
| "epoch": 30, | |
| "optimizer": "Adam", | |
| "save_interval": 1, | |
| "pretrain": "pretrained/fast_small_in1k_epoch_299.pth" | |
| }, | |
| "test_cfg": { | |
| "result_path": "outputs/submit_ctw/", | |
| "min_area": 250, | |
| "min_score": 0.88, | |
| "bbox_type": "rect" | |
| } | |
| } | |