Hugging Face's logo

Raghavan
/
fast_model_config_files

Model card Files Community
fast_model_config_files
File size: 1,519 Bytes 
2e7c6fe
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
model = dict(
    type='FAST',
    backbone=dict(
        type='fast_backbone',
        config='config/fast/nas-configs/fast_small.config'
    ),
    neck=dict(
        type='fast_neck',
        config='config/fast/nas-configs/fast_small.config'
    ),
    detection_head=dict(
        type='fast_head',
        config='config/fast/nas-configs/fast_small.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='test',
        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='pretrained/fast_small_in1k_epoch_299.pth'
    # https://github.com/czczup/FAST/releases/download/release/fast_small_in1k_epoch_299.pth
)
test_cfg = dict(
    result_path='outputs/submit_ctw/',
    min_area=250,
    min_score=0.88,
    bbox_type='rect',
)