model_name: missioncore-m48s-mask-grounding-dino-shadow
results_dir: /workspace/probe/results
wandb:
  enable: false
model:
  backbone: swin_tiny_224_1k
  num_queries: 900
  num_feature_levels: 4
  dec_layers: 6
  enc_layers: 6
  dim_feedforward: 2048
  dropout_ratio: 0.0
  log_scale: auto
  class_embed_bias: true
  num_region_queries: 100
  loss_types:
    - labels
    - boxes
    - masks
    - rela
dataset:
  infer_data_sources:
    image_dir: /workspace/probe/input
    data_type: OD
    captions:
      - adult person
      - child
      - dog
      - bicycle
      - kick scooter
      - stroller
      - shopping cart
      - car
      - bus
      - truck
      - motorcycle
      - trash bin
      - bollard
      - post
      - concrete hemisphere
      - traffic cone
      - road sign
      - temporary barrier
  batch_size: 1
  workers: 1
  pin_memory: true
  max_labels: 80
  has_mask: true
inference:
  trt_engine: /workspace/model/mask_grounding_dino_swin_tiny_research_fp16.engine
  results_dir: /workspace/probe/results
  batch_size: -1
  input_width: 960
  input_height: 544
  conf_threshold: 0.25
  ioi_threshold: 0.5
  nms_threshold: 0.2
  text_threshold: 0.3
  outline_width: 3
gen_trt_engine:
  onnx_file: /workspace/model/mask_grounding_dino_epoch_021.onnx
  trt_engine: /workspace/model/mask_grounding_dino_swin_tiny_research_fp16.engine
  results_dir: /workspace/model/engine-build
  batch_size: -1
  verbose: false
  tensorrt:
    workspace_size: 8192
    min_batch_size: 1
    opt_batch_size: 1
    max_batch_size: 1
    data_type: FP16