| model = dict( |
| type='FAST', |
| backbone=dict( |
| type='fast_backbone', |
| config='config/fast/nas-configs/fast_tiny.config' |
| ), |
| neck=dict( |
| type='fast_neck', |
| config='config/fast/nas-configs/fast_tiny.config' |
| ), |
| detection_head=dict( |
| type='fast_head', |
| config='config/fast/nas-configs/fast_tiny.config', |
| pooling_size=9, |
| loss_text=dict( |
| type='DiceLoss', |
| loss_weight=0.5 |
| ), |
| loss_kernel=dict( |
| type='DiceLoss', |
| loss_weight=1.0 |
| ), |
| loss_emb=dict( |
| type='EmbLoss_v1', |
| feature_dim=4, |
| loss_weight=0.25 |
| ) |
| ) |
| ) |
| repeat_times = 10 |
| data = dict( |
| batch_size=16, |
| train=dict( |
| type='FAST_IC17MLT', |
| split='train', |
| is_transform=True, |
| img_size=640, |
| short_size=640, |
| pooling_size=9, |
| read_type='cv2', |
| repeat_times=repeat_times, |
| ), |
| test=dict( |
| type='FAST_IC17MLT', |
| split='valid', |
| short_size=640, |
| read_type='cv2' |
| ) |
| ) |
| train_cfg = dict( |
| lr=1e-3, |
| schedule='polylr', |
| epoch=300 // repeat_times, |
| optimizer='Adam', |
| save_interval=10 // repeat_times, |
| pretrain='/Users/eaxxkra/Downloads/fast_tiny_ic17mlt_640.pth' |
| |
| ) |
| test_cfg = dict( |
| result_path='outputs/submit_ctw/', |
| min_area=250, |
| min_score=0.88, |
| bbox_type='rect', |
| ) |
|
|
