File size: 1,737 Bytes
ce4ff03 8fefaa1 ce4ff03 8fefaa1 ce4ff03 8fefaa1 ce4ff03 8fefaa1 ce4ff03 8fefaa1 ce4ff03 |
1 2 3 4 5 6 7 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 |
{
"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"
}
}
|