File size: 1,183 Bytes
9b92c75 | 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 | model:
num_classes: 80
input_size: 640
stem_channels: 64
backbone_channels: [96, 192, 384, 512]
backbone_depths: [2, 3, 6, 3]
hidden_dim: 384
fpn_depth: 2
latent_count: 64
latent_pool_sizes: [12, 6, 3]
latent_layers: 2
decoder_layers: 8
num_queries: 300
num_heads: 8
local_points: 4
dropout: 0.0
dense_aux: true
loss:
cost_class: 2.0
cost_bbox: 5.0
cost_giou: 2.0
weight_class: 2.0
weight_bbox: 5.0
weight_giou: 2.0
weight_dense: 1.0
focal_alpha: 0.25
focal_gamma: 2.0
aux_weight: 1.0
dense_topk: 5
train:
epochs: 150
batch_size: 16
eval_batch_size: 8
workers: 8
prefetch_factor: 4
lr: 0.0002
backbone_lr: 0.0001
min_lr_ratio: 0.05
weight_decay: 0.05
warmup_steps: 1500
clip_grad_norm: 0.1
amp: true
amp_dtype: bfloat16
channels_last: false
compile: false
ema_decay: 0.9998
seed: 42
eval_every: 1
print_freq: 50
data:
train_image_dir: train2017
train_annotations: annotations/instances_train2017.json
val_image_dir: val2017
val_annotations: annotations/instances_val2017.json
hflip_prob: 0.5
scale_range: [0.65, 1.0]
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
|