config.yaml

# BITFSD-Annotator Configuration

# ── Common (both backends need these) ─────────────────────────────────
common:
  data_dir: data

# ── PyTorch backend only ──────────────────────────────────────────────
pytorch:
  model_config: /home/as/vllm/pc_yolo/OpenPCDet/tools/cfgs/custom_models/centerpoint_cone.yaml
  checkpoint: /home/as/vllm/pc_yolo/OpenPCDet/output/custom_models/centerpoint_cone_final/default/ckpt/checkpoint_epoch_120.pth

# ── TensorRT backend only ─────────────────────────────────────────────
tensorrt:
  trt_config: /home/as/vllm/pc_yolo/test_ws/src/centerpoint/config/centerpoint_v008.yaml
  engine: /home/as/vllm/pc_yolo/test_ws/src/centerpoint/weights/v008.engine
  vfe_weights: /home/as/vllm/pc_yolo/test_ws/src/centerpoint/weights/vfe_weights_v008.bin

# ── Inference settings ────────────────────────────────────────────────
inference:
  backend: tensorrt       # "pytorch" | "tensorrt"
  score_thresh: 0.35

# ── SOTIF quality gate ────────────────────────────────────────────────
sotif:
  min_points_discard: 5
  min_points_hard: 8