config.json

{
  "batch_size": 2,
  "gradient_accumulation_steps": 4,
  "epochs": 200,
  "lr": 5e-05,
  "backbone_lr": 1e-05,
  "weight_decay": 0.0001,
  "backbone_type": "resnet50",
  "use_pretrained": true,
  "freeze_backbone_epochs": 5,
  "use_gradient_checkpointing": true,
  "image_size": [
    1920,
    1080
  ],
  "num_classes": 11,
  "merge_banknotes": true,
  "train_ratio": 0.85,
  "test_ratio": 0.15,
  "fpn_channels": 256,
  "strides": [
    8,
    16,
    32,
    64,
    128
  ],
  "use_native_resolution": true,
  "focal_alpha": 0.5,
  "focal_gamma": 2.5,
  "lambda_reg": 2.0,
  "lambda_ctr": 1.0,
  "nms_threshold": 0.4,
  "score_threshold": 0.35,
  "max_detections": 100,
  "patience": 30,
  "grad_clip": 1.0,
  "warmup_epochs": 10,
  "hf_dataset": "format37/russian-rubles-banknotes",
  "hf_model_repo": "format37/fcos-banknotes-detector"
}